adapta 3.3.1a500.dev4__tar.gz → 3.3.2a501.dev4__tar.gz

{adapta-3.3.1a500.dev4 → adapta-3.3.2a501.dev4}/PKG-INFO

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

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

{adapta-3.3.1a500.dev4 → adapta-3.3.2a501.dev4}/adapta/ml/mlflow/_client.py

@@ -80,7 +80,7 @@ class MlflowBasicClient:
         """
         return self._client.get_model_version_by_alias(model_name, alias)
 
-    def _get_artifact_repo_backported(self, run_id) -> mlflow.store.artifact_repo.ArtifactRepository:
+    def _get_artifact_repo_backported(self, run_id: str) -> mlflow.store.artifact_repo.ArtifactRepository:
         run = self._client.get_run(run_id)
 
         artifact_uri = (
@@ -137,6 +137,26 @@ class MlflowBasicClient:
             version=model_version,
         )
 
+    def log_dict(self, artifact: dict, artifact_path: str, run_id: str):
+        """
+        inherited the logging dictionary in Mlflow
+
+        :param artifact: dictionary to log
+        :param artifact_path: artifact path
+        :param run_id: run id
+        """
+        self._client.log_dict(run_id=run_id, dictionary=artifact, artifact_file=artifact_path)
+
+    def log_metric(self, run_id: str, metric_name: str, metric_value: float):
+        """
+        inherited the logging metric in Mlflow
+
+        :param run_id: run id
+        :param metric_name: metric name
+        :param metric_value: metric value
+        """
+        self._client.log_metric(run_id=run_id, key=metric_name, value=metric_value)
+
     @staticmethod
     def load_model_by_name(model_name: str, stage_or_version: str) -> PyFuncModel:
         """

{adapta-3.3.1a500.dev4 → adapta-3.3.2a501.dev4}/adapta/ml/mlflow/_functions.py

@@ -18,7 +18,7 @@ import configparser
 import importlib
 import pathlib
 import tempfile
-from typing import Optional, Dict, Any
+from typing import Optional, Dict, Any, Literal
 
 import mlflow
 from mlflow.pyfunc import PythonModel
@@ -52,12 +52,13 @@ def register_mlflow_model(
     model_name: str,
     experiment: str,
     run_name: str = None,
-    transition_to_stage: str = None,
-    version_alias: str = None,
+    transition_to_stage: Optional[Literal["Staging", "Production"]] = None,
+    parent_run_id: Optional[str] = None,
+    version_alias: Optional[str] = None,
     metrics: Optional[Dict[str, float]] = None,
     model_params: Optional[Dict[str, Any]] = None,
     artifacts_to_log: Dict[str, str] = None,
-):
+) -> str:
     """Registers mlflow model
 
     :param model: Machine learning model to register
@@ -65,11 +66,14 @@ def register_mlflow_model(
     :param model_name: Name of Mlflow model
     :param experiment: Name of Mlflow experiment
     :param run_name: Name of Mlflow run
+    :param parent_run_id: Parent run id
     :param transition_to_stage: Whether to transition to stage
     :param version_alias: Alias to assign to model
     :param metrics: Metrics to log
     :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
     """
     assert transition_to_stage in [None, "Staging", "Production"]
 
@@ -98,7 +102,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):
+    with mlflow.start_run(nested=True, run_name=run_name, parent_run_id=parent_run_id) as run:
         mlflow.pyfunc.log_model(
             artifact_path="mlflow_model",
             python_model=_MlflowMachineLearningModel(),
@@ -123,3 +127,5 @@ def register_mlflow_model(
                 stage=transition_to_stage,
                 model_version=version,
             )
+
+        return run.info.run_id

{adapta-3.3.1a500.dev4 → adapta-3.3.2a501.dev4}/pyproject.toml

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

@@ -1 +0,0 @@
-__version__ = 'v3.3.1a500.dev4'