diff --git a/forecasttools/__init__.py b/forecasttools/__init__.py index 57231ad..1499c2b 100644 --- a/forecasttools/__init__.py +++ b/forecasttools/__init__.py @@ -4,7 +4,7 @@ import polars as pl from forecasttools.daily_to_epiweekly import df_aggregate_to_epiweekly -from forecasttools.idata_to_tidy import convert_idata_forecast_to_tidydraws +from forecasttools.idata_to_tidy import convert_inference_data_to_tidydraws from forecasttools.idata_w_dates_to_df import ( add_time_coords_to_idata_dimension, add_time_coords_to_idata_dimensions, @@ -96,5 +96,5 @@ "generate_time_range_for_dim", "validate_iter_has_expected_types", "ensure_listlike", - "convert_idata_forecast_to_tidydraws" + "convert_inference_data_to_tidydraws" ] diff --git a/forecasttools/idata_to_tidy.py b/forecasttools/idata_to_tidy.py index c303eda..ff3a548 100644 --- a/forecasttools/idata_to_tidy.py +++ b/forecasttools/idata_to_tidy.py @@ -11,7 +11,7 @@ import polars as pl -def convert_idata_forecast_to_tidydraws( +def convert_inference_data_to_tidydraws( idata: az.InferenceData, groups: list[str] ) -> dict[str, pl.DataFrame]: diff --git a/notebooks/pyrenew_dates_to_tidy.qmd b/notebooks/pyrenew_dates_to_tidy.qmd index 85e5f02..d827437 100644 --- a/notebooks/pyrenew_dates_to_tidy.qmd +++ b/notebooks/pyrenew_dates_to_tidy.qmd @@ -33,7 +33,7 @@ __Define Groups To Save And Convert__ #| echo: true pyrenew_groups = ["posterior_predictive"] -tidy_usable_groups = forecasttools.convert_idata_forecast_to_tidydraws( +tidy_usable_groups = forecasttools.convert_inference_data_to_tidydraws( idata=pyrenew_idata, groups=pyrenew_groups ) @@ -88,7 +88,7 @@ __Again Convert The Dated Pyrenew InferenceData To Tidy-Usable__ ```{python} pyrenew_groups = ["posterior_predictive"] -tidy_usable_groups_w_dates = forecasttools.convert_idata_forecast_to_tidydraws( +tidy_usable_groups_w_dates = forecasttools.convert_inference_data_to_tidydraws( idata=pyrenew_idata_w_dates, groups=pyrenew_groups )