add minor numpy to python change in trajectories to quant; example fi…

…le additions
CDCgov · Oct 8, 2024 · 2a60571 · 2a60571
1 parent d61adb3
commit 2a60571
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 5 deletions.
diff --git a/forecasttools/to_flusight.py b/forecasttools/to_flusight.py
@@ -12,7 +12,7 @@ def calculate_epidate(date):
 def get_flusight_target_end_dates(
     reference_date: str, horizons=None
 ) -> pl.DataFrame:
-    # default horizons
+    # set default horizons in case of no specification
     if horizons is None:
         horizons = list(range(-1, 4))
     reference_date_dt = datetime.strptime(reference_date, "%Y-%m-%d")

diff --git a/forecasttools/trajectories_to_quantiles.py b/forecasttools/trajectories_to_quantiles.py
@@ -14,7 +14,9 @@ def trajectories_to_quantiles(
     # set default quantiles
     if quantiles is None:
         quantiles = (
-            [0.01, 0.025] + list(np.arange(0.05, 1.0, 0.05)) + [0.975, 0.99]
+            [0.01, 0.025]
+            + [0.05 * elt for elt in range(1, 20)]
+            + [0.975, 0.99]
         )
     # group trajectories based on timepoint_cols and id_cols
     group_cols = (

diff --git a/notebooks/example_fixes.py b/notebooks/example_fixes.py
@@ -9,10 +9,31 @@
 
 # %% IMPORTS
 
+
 import polars as pl
 
 import forecasttools
 
+# %% EXAMINING FILE PATH BEING RETURNED
+
+# example dataframe and location table
+example_dict = {
+    "date": ["2024-10-08", "2024-10-08", "2024-10-08"],
+    "location": ["AL", "AK", "US"],
+}
+example_df = pl.from_dict(example_dict)
+
+# call loc_abbr_to_flusight_code()
+recoded_df = forecasttools.loc_abbr_to_flusight_code(
+    df=example_df, location_col="location"
+)
+
+# received warning (original implementation)
+# MapWithoutReturnDtypeWarning: Calling `map_elements` without specifying `return_dtype` can lead to unpredictable results. Specify `return_dtype` to silence this warning.
+
+# no longer receives warning after using expr.replace()
+
+
 # %% USING POLARS EXPR.REPLACE()
 
 # example dataframe and location table
@@ -22,15 +43,12 @@
 }
 example_df = pl.from_dict(example_dict)
 loc_table = forecasttools.location_table
-print(example_df)
-print(loc_table)
 
 # replace "location" values with codes from loc_table
 new_df = example_df.with_columns(
     location=pl.col("location").replace(
         old=loc_table["short_name"], new=loc_table["location_code"]
     )
 )
-print(new_df)
 
 # %%