check for invalid groups

CDCgov · Jan 29, 2025 · 9452acd · 9452acd
1 parent 0ad7c24
commit 9452acd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/forecasttools/idata_to_tidy.py b/forecasttools/idata_to_tidy.py
@@ -37,8 +37,14 @@ def convert_inference_data_to_tidydraws(
         A dictionary of groups from the idata
         for use with the tidybayes API.
     """
+    available_groups = list(idata.groups())
     if groups is None:
-        groups = list(idata.groups())
+        groups = available_groups
+    else:
+        invalid_groups = [group for group in groups if group not in available_groups]
+        if invalid_groups:
+            raise ValueError(f"Invalid groups provided: {invalid_groups}. Available groups: {available_groups}")
+
     idata_df = pl.DataFrame(idata.to_dataframe())
 
     tidy_dfs = {