fix: proper guard against neither df nor output_spec specified in…

… `get_value_col_str_name`
Aarhus-Psychiatry-Research · Feb 9, 2023 · aeef039 · aeef039
1 parent 45245a3
commit aeef039
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/timeseriesflattener/multi_index_handling.py b/src/timeseriesflattener/multi_index_handling.py
@@ -92,7 +92,7 @@ def get_value_col_str_name(
         Args:
             df (pd.DataFrame): Dataframe to get value column name from.
             output_spec (TemporalSpec): Output specification"""
-        if df or output_spec:
+        if df is not None or output_spec is not None:
             return (
                 df["value"].columns.tolist()  # type: ignore
                 if isinstance(df.columns, pd.MultiIndex)  # type: ignore