refactor: standardize str to datetime

tests/conftest.py

@@ -37,7 +37,6 @@ def path_example_img():
 @pytest.fixture
 def example_forecast_df(path_example_forecast_csv):
     df = load_file(path_example_forecast_csv, file_type="csv")
-    df = str_to_datetime(df, date_column="ds", date_fmt="%Y-%m-%d %H:%M:%S")
     return df
 
 

tests/test_forecast.py

@@ -5,6 +5,7 @@
 from vaxstats.cli import main
 from vaxstats.forecast import run_forecasting
 from vaxstats.io import load_file
+from vaxstats.utils import str_to_datetime
 
 
 def test_arima(path_example_prepped_csv):
@@ -15,6 +16,7 @@ def test_arima(path_example_prepped_csv):
         "method": "CSS-ML",  # CSS-ML, ML, CSS
     }
     df = load_file(path_example_prepped_csv, "csv")
+    df = str_to_datetime(df, date_column="ds", date_fmt="%Y-%m-%d %H:%M:%S")
     baseline_days = 7.0
     baseline_hours = 24 * baseline_days
     df = run_forecasting(

tests/test_stats.py

@@ -5,7 +5,7 @@
 from vaxstats.analysis.hourly import add_hourly_thresholds, calculate_hourly_stats
 from vaxstats.analysis.residual import add_residuals_col, get_residual_bounds
 from vaxstats.analysis.stats import get_column_stat, get_column_stats
-from vaxstats.utils import get_baseline_df
+from vaxstats.utils import get_baseline_df, str_to_datetime
 
 
 def test_get_column_stat(example_forecast_df):
@@ -44,8 +44,9 @@ def test_residual_bounds(example_forecast_df_baseline):
 
 
 def test_calculate_hourly_stats(example_forecast_df_baseline):
+    df = example_forecast_df_baseline
     hourly_stats = calculate_hourly_stats(
-        example_forecast_df_baseline,
+        df,
         data_column="y",
         pred_column="y_hat",
         date_column="ds",
@@ -77,6 +78,7 @@ def test_calculate_hourly_stats(example_forecast_df_baseline):
 def test_calculate_thresholds(example_forecast_df, baseline_hours):
     df = example_forecast_df
     df = add_residuals_col(df)
+    df = str_to_datetime(df, date_column="ds", date_fmt="%Y-%m-%d %H:%M:%S")
     df_baseline = get_baseline_df(df, baseline=baseline_hours)
 
     residual_lower, residual_upper = get_residual_bounds(df_baseline)
@@ -103,6 +105,7 @@ def test_calculate_thresholds(example_forecast_df, baseline_hours):
 def test_detect_fever_hypothermia(example_forecast_df, baseline_hours):
     df = example_forecast_df
     df = add_residuals_col(df)
+    df = str_to_datetime(df, date_column="ds", date_fmt="%Y-%m-%d %H:%M:%S")
     hourly_stats, residual_bounds = detect_fever_hypothermia(
         df, baseline=baseline_hours
     )
@@ -126,6 +129,7 @@ def test_detect_fever_hypothermia(example_forecast_df, baseline_hours):
 def test_get_all_stats(example_forecast_df, baseline_hours):
     df = example_forecast_df
     df = add_residuals_col(df)
+    df = str_to_datetime(df, date_column="ds", date_fmt="%Y-%m-%d %H:%M:%S")
 
     results = run_analysis(
         df,

vaxstats/analysis/hourly.py

@@ -21,7 +21,7 @@ def calculate_hourly_stats(
     """
     return (
         df.with_columns(pl.col(date_column).dt.truncate("1h").alias("hour"))
-        .groupby("hour")
+        .group_by("hour")
         .agg(
             pl.col(data_column).median().alias("y_median"),
             pl.col(pred_column).median().alias("y_hat_median"),

vaxstats/forecast.py

@@ -7,7 +7,7 @@
 from .analysis.residual import add_residuals_col
 from .io import load_file
 from .log import run_with_progress_logging
-from .utils import split_df
+from .utils import split_df, str_to_datetime
 
 
 def run_forecasting(
@@ -81,6 +81,7 @@ def run_forecasting(
 def cli_forecast(args, sf_model_args=(), sf_model_kwargs={}):
     # Load the input file
     df = load_file(args.file_path)
+    df = str_to_datetime(df, date_column="ds", date_fmt="%Y-%m-%d %H:%M:%S")
 
     # Import the forecasting model class dynamically
     module_name, class_name = args.sf_model.rsplit(".", 1)