adapta 3.3.2a501.dev6__tar.gz → 3.3.2a501.dev10__tar.gz

{adapta-3.3.2a501.dev6 → adapta-3.3.2a501.dev10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: adapta
-Version: 3.3.2a501.dev6
+Version: 3.3.2a501.dev10
 Summary: Logging, data connectors, monitoring, secret handling and general lifehacks to make data people lives easier.
 License: Apache 2.0
 Author: ECCO Sneaks & Data

adapta-3.3.2a501.dev10/adapta/_version.py

@@ -0,0 +1 @@
+__version__ = 'v3.3.2a501.dev10'

{adapta-3.3.2a501.dev6 → adapta-3.3.2a501.dev10}/adapta/ml/mlflow/_client.py

@@ -157,15 +157,15 @@ class MlflowBasicClient:
         """
         self._client.log_metric(run_id=run_id, key=metric_name, value=metric_value)
 
-    def create_run(self, experiment_name: str, run_name: str = None) -> str:
+    def create_run(self, experiment_id: str, run_name: str = None) -> str:
         """
         inherited the creating run in Mlflow
 
-        :param experiment_name: experiment name
+        :param experiment_id: experiment id
         :param run_name: run name
         :return: run id
         """
-        return self._client.create_run(experiment_id=experiment_name, run_name=run_name).info.run_id
+        return self._client.create_run(experiment_id=experiment_id, run_name=run_name).info.run_id
 
     @staticmethod
     def load_model_by_name(model_name: str, stage_or_version: str) -> PyFuncModel:
@@ -174,7 +174,6 @@ class MlflowBasicClient:
         """
         return mlflow.pyfunc.load_model(f"models:/{model_name}/{stage_or_version}")
 
-
     @staticmethod
     def load_model_by_uri(model_uri: str) -> PyFuncModel:
         """

{adapta-3.3.2a501.dev6 → adapta-3.3.2a501.dev10}/adapta/ml/mlflow/_functions.py

@@ -51,7 +51,8 @@ def register_mlflow_model(
     mlflow_client: MlflowBasicClient,
     model_name: str,
     experiment: str,
-    run_name: str = None,
+    run_name: Optional[str] = None,
+    run_id: Optional[str] = None,
     transition_to_stage: Optional[Literal["Staging", "Production"]] = None,
     parent_run_id: Optional[str] = None,
     version_alias: Optional[str] = None,
@@ -65,7 +66,8 @@ def register_mlflow_model(
     :param mlflow_client: Mlflow client
     :param model_name: Name of Mlflow model
     :param experiment: Name of Mlflow experiment
-    :param run_name: Name of Mlflow run
+    :param run_name: Name of Mlflow run (only used if run_id is None)
+    :param run_id: Run id
     :param parent_run_id: Parent run id
     :param transition_to_stage: Whether to transition to stage
     :param version_alias: Alias to assign to model
@@ -73,7 +75,8 @@ def register_mlflow_model(
     :param model_params: Model hyperparameters to log
     :param artifacts_to_log: Additional artifacts to log
 
-    :return: Run id of the newly created run for registering the model
+    :return: Run id of the newly created run for registering the model.
+    If run_id is provided, it will be the same as run_id
     """
     assert transition_to_stage in [None, "Staging", "Production"]
 
@@ -102,7 +105,7 @@ def register_mlflow_model(
             raise ValueError('Artifact names "model" and "config" are reserved for internal usage')
         artifacts.update(artifacts_to_log)
 
-    with mlflow.start_run(nested=True, run_name=run_name, parent_run_id=parent_run_id) as run:
+    with mlflow.start_run(nested=True, run_name=run_name, run_id=run_id, parent_run_id=parent_run_id) as run:
         mlflow.pyfunc.log_model(
             artifact_path="mlflow_model",
             python_model=_MlflowMachineLearningModel(),

{adapta-3.3.2a501.dev6 → adapta-3.3.2a501.dev10}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "adapta"
-version = "v3.3.2a501.dev6"
+version = "v3.3.2a501.dev10"
 description = "Logging, data connectors, monitoring, secret handling and general lifehacks to make data people lives easier."
 authors = ["ECCO Sneaks & Data <esdsupport@ecco.com>"]
 maintainers = ['GZU <gzu@ecco.com>', 'JRB <ext-jrb@ecco.com>']

adapta-3.3.2a501.dev6/adapta/_version.py

@@ -1 +0,0 @@
-__version__ = 'v3.3.2a501.dev6'