Skip to content

Commit

Permalink
rename daily_to_epiweeks to df_daily_dates_to_epiweeks
Browse files Browse the repository at this point in the history
  • Loading branch information
AFg6K7h4fhy2 committed Oct 15, 2024
1 parent be30f78 commit 8d5d233
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions forecasttools/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import arviz as az
import polars as pl

from .daily_to_epiweekly import daily_to_epiweekly
from .df_daily_dates_to_epiweeks import df_daily_dates_to_epiweeks
from .idata_to_df_w_dates import forecast_as_df_with_dates
from .recode_locations import loc_abbr_to_flusight_code
from .to_flusight import get_flusight_table
Expand Down Expand Up @@ -43,7 +43,7 @@
"nhsn_flu_forecast",
"forecast_as_df_with_dates",
"trajectories_to_quantiles",
"daily_to_epiweekly",
"df_daily_dates_to_epiweeks",
"loc_abbr_to_flusight_code",
"get_flusight_table",
]
2 changes: 1 addition & 1 deletion forecasttools/daily_to_epiweekly.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ def calculate_epi_week_and_year(date: str):
return epiweek_df_struct


def daily_to_epiweekly(
def df_daily_dates_to_epiweeks(
forecast_df: pl.DataFrame,
value_col: str = "value",
date_col: str = "date",
Expand Down
2 changes: 1 addition & 1 deletion notebooks/flusight_from_idata.qmd
Original file line number Diff line number Diff line change
Expand Up @@ -195,7 +195,7 @@ Format to epiweeks:

```{python}
forecast_df = forecasttools.daily_to_epiweekly(
forecast_df = forecasttools.df_daily_dates_to_epiweeks(
forecast_df,
value_col = "hosp",
id_cols = [".draw", "location"],
Expand Down

0 comments on commit 8d5d233

Please sign in to comment.