openstef 3.3.11__py3-none-any.whl → 3.3.13__py3-none-any.whl

openstef/model/serializer.py

@@ -27,6 +27,11 @@ class MLflowSerializer:
         self.logger = structlog.get_logger(self.__class__.__name__)
         mlflow.set_tracking_uri(mlflow_tracking_uri)
         self.logger.debug(f"MLflow tracking uri at init= {mlflow_tracking_uri}")
+        self.experiment_name_prefix = (
+            os.environ["DATABRICKS_WORKSPACE_PATH"]
+            if "DATABRICKS_WORKSPACE_PATH" in os.environ
+            else ""
+        )
 
     def save_model(
         self,
@@ -39,7 +44,9 @@ class MLflowSerializer:
         **kwargs,
     ) -> None:
         """Save sklearn compatible model to MLFlow."""
-        mlflow.set_experiment(experiment_name=experiment_name)
+        mlflow.set_experiment(
+            experiment_name=self.experiment_name_prefix + experiment_name
+        )
         with mlflow.start_run(run_name=experiment_name):
             self._log_model_with_mlflow(
                 model=model,
@@ -69,7 +76,7 @@ class MLflowSerializer:
         """
         # Get previous run id
         models_df = self._find_models(
-            experiment_name, max_results=1
+            self.experiment_name_prefix + experiment_name, max_results=1
         )  # returns latest model
         if not models_df.empty:
             previous_run_id = models_df["run_id"][
@@ -143,7 +150,7 @@ class MLflowSerializer:
         """
         try:
             models_df = self._find_models(
-                experiment_name, max_results=1
+                self.experiment_name_prefix + experiment_name, max_results=1
             )  # return the latest finished run of the model
             if not models_df.empty:
                 latest_run = models_df.iloc[0]  # Use .iloc[0] to only get latest run
@@ -177,7 +184,9 @@ class MLflowSerializer:
         if hyperparameter_optimization_only:
             filter_string += " AND tags.phase = 'Hyperparameter_opt'"
         models_df = self._find_models(
-            experiment_name, max_results=1, filter_string=filter_string
+            self.experiment_name_prefix + experiment_name,
+            max_results=1,
+            filter_string=filter_string,
         )
         if not models_df.empty:
             run = models_df.iloc[0]  # Use .iloc[0] to only get latest run
@@ -283,7 +292,9 @@ class MLflowSerializer:
             raise ValueError(
                 f"Max models to keep should be greater than 1! Received: {max_n_models}"
             )
-        previous_runs = self._find_models(experiment_name=experiment_name)
+        previous_runs = self._find_models(
+            experiment_name=self.experiment_name_prefix + experiment_name
+        )
         if len(previous_runs) > max_n_models:
             self.logger.debug(
                 f"Going to delete old models. {len(previous_runs)} > {max_n_models}"

{openstef-3.3.11.dist-info → openstef-3.3.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: openstef
-Version: 3.3.11
+Version: 3.3.13
 Summary: Open short term energy forecaster
 Home-page: https://github.com/OpenSTEF/openstef
 Author: Alliander N.V
@@ -30,7 +30,7 @@ Requires-Dist: scikit-learn ~=1.3
 Requires-Dist: scipy ~=1.10
 Requires-Dist: statsmodels ~=0.13.5
 Requires-Dist: structlog ~=23.1
-Requires-Dist: xgboost ~=1.7
+Requires-Dist: xgboost ~=2.0
 Provides-Extra: proloaf
 Requires-Dist: proloaf ==0.2.0 ; extra == 'proloaf'
 Requires-Dist: torch ==1.10.0 ; extra == 'proloaf'

{openstef-3.3.11.dist-info → openstef-3.3.13.dist-info}/RECORD

@@ -36,7 +36,7 @@ openstef/model/fallback.py,sha256=VV9ehgnoMZtWzqKk9H1t8wnERFh5CyC4TvDIuRP_ZDI,28
 openstef/model/model_creator.py,sha256=U1Lw4HFyajfxQ2o5lEnCxnmRC62DEu5PBHXrm_jnEJU,5582
 openstef/model/objective.py,sha256=qBZGij4P2uZXFt-1x4MxCkIRmpVY3TJoo4Mycxu9dO8,15433
 openstef/model/objective_creator.py,sha256=OiPPFSiSu7z9K_983ib5iqhhu6_9tt7iyTyKNZ2Iz68,2057
-openstef/model/serializer.py,sha256=hudTi6lWI0_B3Duald37ERlNdGNNrPEKI8jdOCXdD9E,16536
+openstef/model/serializer.py,sha256=Q7pKkA-K1bszER_tkOWhvwTQsj6-qZoX5zaXIemRKGs,16934
 openstef/model/standard_deviation_generator.py,sha256=ioQHEIMND_3cjK_-WY-Serx39DaSrA-a2XozuBSNjMM,2912
 openstef/model/metamodels/__init__.py,sha256=bIyGTSA4V5VoOLTwdaiJJAnozmpSzvQooVYlsf8H4eU,163
 openstef/model/metamodels/grouped_regressor.py,sha256=yMN_a6TnQSyFaqlB_6Nifq-ydpb5hs6w_b97IaBbHj4,8337
@@ -85,8 +85,8 @@ openstef/tasks/utils/predictionjobloop.py,sha256=Ysy3zF5lzPMz_asYDKeF5m0qgVT3tCt
 openstef/tasks/utils/taskcontext.py,sha256=yI6TntOkZcW8JiNVuw4uJIigEBL0_iIrkPklF4ZeCX4,5401
 openstef/validation/__init__.py,sha256=bIyGTSA4V5VoOLTwdaiJJAnozmpSzvQooVYlsf8H4eU,163
 openstef/validation/validation.py,sha256=Mhgizxn-zBGjf5m6WdTLO-WAppFJCPQGD29OsmmelAU,10347
-openstef-3.3.11.dist-info/LICENSE,sha256=7Pm2fWFFHHUG5lDHed1vl5CjzxObIXQglnYsEdtjo_k,14907
-openstef-3.3.11.dist-info/METADATA,sha256=NR8YRke9g4kvMQkod8sPaurB0fsj9Xn7QYp9PwITRVk,7840
-openstef-3.3.11.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-openstef-3.3.11.dist-info/top_level.txt,sha256=kD0H4PqrQoncZ957FvqwfBxa89kTrun4Z_RAPs_HhLs,9
-openstef-3.3.11.dist-info/RECORD,,
+openstef-3.3.13.dist-info/LICENSE,sha256=7Pm2fWFFHHUG5lDHed1vl5CjzxObIXQglnYsEdtjo_k,14907
+openstef-3.3.13.dist-info/METADATA,sha256=-UyuETPqm_m3Rup7hzqEfiR0kc6X6F_pd9dPzhIl8Vg,7840
+openstef-3.3.13.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+openstef-3.3.13.dist-info/top_level.txt,sha256=kD0H4PqrQoncZ957FvqwfBxa89kTrun4Z_RAPs_HhLs,9
+openstef-3.3.13.dist-info/RECORD,,