Update src/MEDS_tabular_automl/generate_summarized_reps.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
mmcdermott · May 27, 2024 · 7fdc37d · 7fdc37d
1 parent c8ca3bb
commit 7fdc37d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/MEDS_tabular_automl/generate_summarized_reps.py b/src/MEDS_tabular_automl/generate_summarized_reps.py
@@ -18,7 +18,8 @@
 
 
 def time_aggd_col_alias_fntr(window_size: str, agg: str) -> Callable[[str], str]:
-    assert agg is not None, "agg must be provided"
+    if agg is None:
+        raise ValueError("Aggregation type 'agg' must be provided")
 
     def f(c: str) -> str:
         return "/".join([window_size] + c.split("/") + [agg])