tweak functions, test

cal-itp · Nov 4, 2024 · ac4c409 · ac4c409
1 parent ed23eba
commit ac4c409
Show file tree

Hide file tree

Showing 6 changed files with 628 additions and 1,597 deletions.
diff --git a/_shared_utils/shared_utils/rt_utils.py b/_shared_utils/shared_utils/rt_utils.py
@@ -561,7 +561,7 @@ def categorize_time_of_day(value: Union[int, dt.datetime]) -> str:
     else:
         return "Evening"
 
-def time_of_day_durations():
+def time_of_day_durations() -> pd.Series:
     """
     Get duration in hours of each time of day classification.
     """