From f63250142716e3adc549ac32389e6b20cb6bb154 Mon Sep 17 00:00:00 2001
From: tiffanychu90 <tiffany.ku@dot.ca.gov>
Date: Wed, 27 Nov 2024 18:25:43 +0000
Subject: [PATCH] (time_series_utils): update to make sure dask.from_map works

---
 rt_segment_speeds/segment_speed_utils/time_series_utils.py | 3 +--
 1 file changed, 1 insertion(+), 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
index 63f8eeb8a..211e18987 100644
--- 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