(time_series_utils): update to make sure dask.from_map works

cal-itp · Dec 3, 2024 · f632501 · f632501
1 parent c6f6dc7
commit f632501
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/rt_segment_speeds/segment_speed_utils/time_series_utils.py b/rt_segment_speeds/segment_speed_utils/time_series_utils.py
@@ -31,10 +31,9 @@ def concatenate_datasets_across_dates(
         paths, 
         date_list, 
         data_type = data_type,
+        get_pandas=get_pandas,
         **kwargs
     )  
-    if get_pandas:
-        df = df.compute()
 
     return df