openstef 3.4.83__tar.gz → 3.4.86__tar.gz

{openstef-3.4.83 → openstef-3.4.86}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openstef
-Version: 3.4.83
+Version: 3.4.86
 Summary: Open short term energy forecaster
 Home-page: https://github.com/OpenSTEF/openstef
 Author: Alliander N.V
@@ -18,7 +18,7 @@ Requires-Dist: holidays==0.21
 Requires-Dist: joblib==1.3.2
 Requires-Dist: lightgbm>=4.0
 Requires-Dist: matplotlib~=3.7
-Requires-Dist: mlflow~=2.3
+Requires-Dist: mlflow-skinny<4.0,>=2.0
 Requires-Dist: networkx~=3.1
 Requires-Dist: numpy<3,>=2
 Requires-Dist: optuna~=3.1
@@ -38,6 +38,8 @@ Provides-Extra: cpu
 Requires-Dist: xgboost-cpu~=2.0; extra == "cpu"
 Provides-Extra: gpu
 Requires-Dist: xgboost; extra == "gpu"
+Provides-Extra: mlflow-full
+Requires-Dist: mlflow<4.0,>=2.0; extra == "mlflow-full"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{openstef-3.4.83 → openstef-3.4.86}/openstef/tasks/create_forecast.py

@@ -82,13 +82,9 @@ def create_forecast_task(
         location=[pj["lat"], pj["lon"]],
         datetime_start=datetime_start,
         datetime_end=datetime_end,
-        market_price=pj.electricity_bidding_zone.value,
+        entsoe_bidding_zone=pj.electricity_bidding_zone.value,
     )
 
-    # Add APX price to the input data for backward compatibility,remove this line when all models are retrained
-    if pj.electricity_bidding_zone == BiddingZone.NL:
-        input_data["APX"] = input_data["day_ahead_electricity_price"]
-
     try:
         # Make forecast with the forecast pipeline
         forecast = create_forecast_pipeline(

{openstef-3.4.83 → openstef-3.4.86}/openstef.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openstef
-Version: 3.4.83
+Version: 3.4.86
 Summary: Open short term energy forecaster
 Home-page: https://github.com/OpenSTEF/openstef
 Author: Alliander N.V
@@ -18,7 +18,7 @@ Requires-Dist: holidays==0.21
 Requires-Dist: joblib==1.3.2
 Requires-Dist: lightgbm>=4.0
 Requires-Dist: matplotlib~=3.7
-Requires-Dist: mlflow~=2.3
+Requires-Dist: mlflow-skinny<4.0,>=2.0
 Requires-Dist: networkx~=3.1
 Requires-Dist: numpy<3,>=2
 Requires-Dist: optuna~=3.1
@@ -38,6 +38,8 @@ Provides-Extra: cpu
 Requires-Dist: xgboost-cpu~=2.0; extra == "cpu"
 Provides-Extra: gpu
 Requires-Dist: xgboost; extra == "gpu"
+Provides-Extra: mlflow-full
+Requires-Dist: mlflow<4.0,>=2.0; extra == "mlflow-full"
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{openstef-3.4.83 → openstef-3.4.86}/openstef.egg-info/requires.txt

@@ -2,7 +2,7 @@ holidays==0.21
 joblib==1.3.2
 lightgbm>=4.0
 matplotlib~=3.7
-mlflow~=2.3
+mlflow-skinny<4.0,>=2.0
 networkx~=3.1
 numpy<3,>=2
 optuna~=3.1
@@ -26,3 +26,6 @@ xgboost-cpu~=2.0
 
 [gpu]
 xgboost
+
+[mlflow-full]
+mlflow<4.0,>=2.0

{openstef-3.4.83 → openstef-3.4.86}/setup.py

@@ -33,7 +33,7 @@ def read_long_description_from_readme():
 
 setup(
     name="openstef",
-    version="3.4.83",
+    version="3.4.86",
     packages=find_packages(include=["openstef", "openstef.*"]),
     description="Open short term energy forecaster",
     long_description=read_long_description_from_readme(),
@@ -64,5 +64,8 @@ setup(
             "xgboost-cpu~=2.0"
         ],  # Add xgboost-cpu to avoid install large xgboost package when not using GPU
         "gpu": ["xgboost"],  # Needed to install xgboost with GPU support in normal case
+        "mlflow-full": [
+            "mlflow>=2.0,<4.0",
+        ],
     },
 )