Merge pull request #1144 from suhr25/fix/cohorts-dict-eval-error-handling

fix: improve error handling for invalid pandas queries in _locate_cohorts()

Author: jonbrenas

malariagen_data/anoph/sample_metadata.py

@@ -1839,7 +1839,12 @@ def _locate_cohorts(*, cohorts, data, min_cohort_size):
         # to pandas queries.
 
         for coh, query in cohorts.items():
-            loc_coh = data.eval(query).values
+            try:
+                loc_coh = data.eval(query).values
+            except Exception as e:
+                raise ValueError(
+                    f"Invalid query for cohort {coh!r}: {query!r}. Error: {e}"
+                ) from e
             coh_dict[coh] = loc_coh
 
     else: