fix: typo resulted in cache breaking

Aarhus-Psychiatry-Research · Nov 10, 2022 · fdd47d7 · fdd47d7
1 parent bc74ae3
commit fdd47d7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/psycop_feature_generation/timeseriesflattener/flattened_dataset.py b/src/psycop_feature_generation/timeseriesflattener/flattened_dataset.py
@@ -327,10 +327,10 @@ def _flatten_temporal_values_to_df(  # noqa pylint: disable=too-many-locals
 
         if verbose:
             msg.good(
-                f"Returning {df.shape[0]} rows of flattened dataframe with {value_cols}",
+                f"Returning {df.shape[0]} rows of flattened dataframe for {output_spec.get_col_str()}",
             )
 
-        return df[[pred_time_uuid_col_name] + output_spec.get_col_str()]
+        return df[[pred_time_uuid_col_name, output_spec.get_col_str()]]
 
     def _generate_values_for_cache_checking(
         self,