cohort update draws from flag configs

amplitude · Aug 6, 2024 · 5b688bd · 5b688bd
1 parent 6ea735e
commit 5b688bd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/experiment/deployment/deployment_runner.rb b/lib/experiment/deployment/deployment_runner.rb
@@ -113,7 +113,8 @@ def download_cohorts(cohort_ids)
 
     def update_cohorts
       @logger.debug('Updating cohorts in storage')
-      download_cohorts(@cohort_storage.cohort_ids)
+      cohort_ids = AmplitudeExperiment.get_all_cohort_ids_from_flags(@flag_config_storage.flag_configs)
+      download_cohorts(cohort_ids)
     end
 
     def delete_unused_cohorts