adapta 3.3.2a501.dev8__tar.gz → 3.3.2a501.dev12__tar.gz

{adapta-3.3.2a501.dev8 → adapta-3.3.2a501.dev12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: adapta
-Version: 3.3.2a501.dev8
+Version: 3.3.2a501.dev12
 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.dev12/adapta/_version.py

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

{adapta-3.3.2a501.dev8 → adapta-3.3.2a501.dev12}/adapta/ml/mlflow/_client.py

@@ -167,6 +167,16 @@ class MlflowBasicClient:
         """
         return self._client.create_run(experiment_id=experiment_id, run_name=run_name).info.run_id
 
+    def set_run_tag(self, key: str, value: any, run_id: str):
+        """
+        inherited the setting run tag in Mlflow
+
+        :param key: tag key
+        :param value: tag value
+        :param run_id: run id
+        """
+        self._client.set_tag(run_id=run_id, key=key, value=value)
+
     @staticmethod
     def load_model_by_name(model_name: str, stage_or_version: str) -> PyFuncModel:
         """
@@ -174,7 +184,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.dev8 → adapta-3.3.2a501.dev12}/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.dev8 → adapta-3.3.2a501.dev12}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "adapta"
-version = "v3.3.2a501.dev8"
+version = "v3.3.2a501.dev12"
 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.dev8/adapta/_version.py

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