Merge pull request #3 from CDCgov/2-create-basic-python-package-struc…

…ture-forecasttools_py Create Basic Python Package Structure, `forecasttools_py`
CDCgov · Sep 25, 2024 · 29e330d · 29e330d
2 parents c9bce73 + d7aebd6
commit 29e330d
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 0 deletions.
diff --git a/forecasttools_py/__init__.py b/forecasttools_py/__init__.py
@@ -0,0 +1,3 @@
+from trajectories_to_quantiles import trajectories_to_quantiles
+
+__all__ = ["trajectories_to_quantiles"]
diff --git a/forecasttools_py/to_flusight_hub.py b/forecasttools_py/to_flusight_hub.py
@@ -0,0 +1,5 @@
+"""
+Get a correctly-formatted set of Flusight `target_end_date` values alongside static FluSight output columns for the given reference date and set of horizons.
+"""
+
+# placeholder
diff --git a/forecasttools_py/trajectories_to_quantiles.py b/forecasttools_py/trajectories_to_quantiles.py
@@ -0,0 +1,8 @@
+"""
+Aggregate individual trajectory timeseries or forecasts to quantile timeseries or forecasts.
+"""
+
+
+# placeholder
+def trajectories_to_quantiles():
+    pass
diff --git a/pyproject.toml b/pyproject.toml
@@ -5,10 +5,13 @@ description = "Python package for common pre- and post-processing operations don
 authors = ["CFA"]
 license = "Apache-2.0"
 readme = "README.md"
+packages = [{include = "forecasttools_py"}]
 
 [tool.poetry.dependencies]
 python = "^3.12"
 pre-commit = "^3.8.0"
+polars = "^1.8.2"
+arviz = "^0.19.0"
 
 
 [build-system]