Skip to content

Commit

Permalink
precision language in docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
AFg6K7h4fhy2 committed Oct 17, 2024
1 parent 9db582b commit 12dd924
Showing 1 changed file with 5 additions and 8 deletions.
13 changes: 5 additions & 8 deletions forecasttools/daily_to_epiweekly.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,12 +31,9 @@ def df_aggregate_to_epiweekly(
strict: bool = False,
) -> pl.DataFrame:
"""
Converts the dates column (daily resolution)
of a Polars dataframe of draws and dates
for a single jurisdiction into
epiweekly and epiyearly columns. This entails
aggregating and replacing the date_col, not
just adding the epiweek and epiyear columns.
Aggregates daily values (e.g.,
hospitalizations) to epiweekly total values.
Add epiweek and epiyear columns.
Parameters
----------
Expand Down Expand Up @@ -70,8 +67,8 @@ def df_aggregate_to_epiweekly(
Returns
-------
pl.DataFrame
A dataframe with date_col replaced by
epiweek and epiyear columns.
A dataframe with value_col aggregated
across epiweek and epiyear.
"""
# check intended df columns are in received df
forecast_df_cols = forecast_df.columns
Expand Down

0 comments on commit 12dd924

Please sign in to comment.