From 9527fb67f8e8f56bec0f3636cc5267d7813e3c3e Mon Sep 17 00:00:00 2001 From: ivelin <ivelin117@gmail.com> Date: Fri, 5 Jul 2024 21:01:31 -0500 Subject: [PATCH] fix: minor updates Signed-off-by: ivelin <ivelin117@gmail.com> --- forecast.sh | 8 +++++++- src/canswim/forecast.py | 3 --- tests/test_cli.py | 3 +++ 3 files changed, 10 insertions(+), 4 deletions(-) create mode 100644 tests/test_cli.py diff --git a/forecast.sh b/forecast.sh index 17a6e35..437f7ca 100755 --- a/forecast.sh +++ b/forecast.sh @@ -22,4 +22,10 @@ set -exv #./canswim.sh forecast --forecast_start_date "2024-04-29" #./canswim.sh forecast --forecast_start_date "2024-05-06" #./canswim.sh forecast --forecast_start_date "2024-05-13" -./canswim.sh forecast --forecast_start_date "2024-05-20" +#./canswim.sh forecast --forecast_start_date "2024-05-20" +#./canswim.sh forecast --forecast_start_date "2024-05-28" +#./canswim.sh forecast --forecast_start_date "2024-06-03" +#./canswim.sh forecast --forecast_start_date "2024-06-10" +#./canswim.sh forecast --forecast_start_date "2024-06-17" +#./canswim.sh forecast --forecast_start_date "2024-06-24" +./canswim.sh forecast --forecast_start_date "2024-07-01" diff --git a/src/canswim/forecast.py b/src/canswim/forecast.py index e512cef..ff9e314 100644 --- a/src/canswim/forecast.py +++ b/src/canswim/forecast.py @@ -5,10 +5,7 @@ import pandas as pd from loguru import logger from canswim.model import CanswimModel -import pandas as pd -from pandas import DataFrame from canswim.hfhub import HFHub -from matplotlib import pyplot from pandas.tseries.offsets import BDay from loguru import logger import os diff --git a/tests/test_cli.py b/tests/test_cli.py new file mode 100644 index 0000000..47d974d --- /dev/null +++ b/tests/test_cli.py @@ -0,0 +1,3 @@ + +# set env vars for each CLI option +# test behavior