Merge pull request #175 from alliander-opensource/bugfix-include-quan…

…tile-models-in-basecase Bugfix include quantile models in basecase
OpenSTEF · Sep 24, 2021 · 2fac389 · 2fac389
2 parents b3dab50 + 51f32a3
commit 2fac389
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/openstf/tasks/create_basecase_forecast.py b/openstf/tasks/create_basecase_forecast.py
@@ -69,7 +69,7 @@ def main():
     taskname = Path(__file__).name.replace(".py", "")
 
     with TaskContext(taskname) as context:
-        model_type = ["xgb", "lgb"]
+        model_type = ["xgb", "xgb_quantile", "lgb"]
 
         PredictionJobLoop(context, model_type=model_type).map(
             create_basecase_forecast_task, context

diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@ def read_long_description_from_readme():
 
 setup(
     name="openstf",
-    version="2.4.0",
+    version="2.4.1",
     packages=find_packages(include=["openstf", "openstf.*"]),
     description="Open short term forcasting",
     long_description=read_long_description_from_readme(),