mend

ArcInstitute · Sep 23, 2024 · 0685fec · 0685fec
1 parent 542a8c2
commit 0685fec
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/screenpro/assays/__init__.py b/screenpro/assays/__init__.py
@@ -344,6 +344,8 @@ def buildPhenotypeData(self, run_name='auto',db_rate_col='pop_doubling', **kwarg
         else:
             raise ValueError('Only `compare_reps` run_name is supported for now!')
 
+        untreated = self.phenotypes[run_name]['config']['untreated']
+
         pdata_list = []
 
         for phenotype_name in self.listPhenotypeScores(run_name=run_name):
@@ -354,7 +356,8 @@ def buildPhenotypeData(self, run_name='auto',db_rate_col='pop_doubling', **kwarg
             #TODO: fix `_calculateGrowthFactor` and `_getTreatmentDoublingRate`
             if db_rate_col:
                 growth_rate_reps = self._calculateGrowthFactor(
-                    untreated = cond_ref, treated = cond_test,
+                    untreated = untreated, 
+                    treated = cond_test, # should be part of "treated" list!
                     db_rate_col = db_rate_col
                 ).query(
                     f'score=="{score_tag}"'