sm_data_ml_utils 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

sm_data_ml_utils/core/config.py

@@ -1,6 +1,6 @@
 from typing import Dict
 
-from pydantic_settings import BaseSettings
+from pydantic import BaseSettings
 
 
 class Settings(BaseSettings):

sm_data_ml_utils/core/databricks_utils.py

@@ -24,6 +24,7 @@ def read_sql(file_path: str) -> str:
     """
     return pathlib.Path(file_path).read_text()
 
+
 def load_yaml(path: str) -> Dict:
     """
     function that loads the settings from the yaml file

sm_data_ml_utils/mlflow_databricks/mlflow_model_utils.py

@@ -7,6 +7,7 @@ from typing import Union
 import mlflow
 import pandas as pd
 import torch
+from joblib import load
 
 from sm_data_ml_utils.core.databricks_utils import get_target_stage_for_env
 from sm_data_ml_utils.core.databricks_utils import load_yaml
@@ -76,6 +77,13 @@ def mlflow_load_artifact(
     if type_of_artifact not in ("joblib", "pkl", "dict", "yaml"):
         raise ValueError("Artifact type not supported")
 
+    if type_of_artifact in ("joblib"):
+        return load(  # noqa: S301
+            mlflow.artifacts.download_artifacts(
+                artifact_uri=f"{artifact_uri}/{artifact_name}"
+            )
+        )
+
     if type_of_artifact in ("joblib", "pkl", "dict"):
         return pd.read_pickle(  # noqa: S301
             mlflow.artifacts.download_artifacts(

{sm_data_ml_utils-0.1.1.dist-info → sm_data_ml_utils-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sm_data_ml_utils
-Version: 0.1.1
+Version: 0.1.3
 Summary: Common Python tools and utilities for ML work
 License: MIT
 Author: Shuming Peh

{sm_data_ml_utils-0.1.1.dist-info → sm_data_ml_utils-0.1.3.dist-info}/RECORD

@@ -1,15 +1,15 @@
 sm_data_ml_utils/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sm_data_ml_utils/core/config.py,sha256=gSdTdbXK-SR4FpRuhaTtR0f7ehyIKPaash463p85BTk,702
-sm_data_ml_utils/core/databricks_utils.py,sha256=chpT8-rugsL_i4sDTi0XJqh3RGqEbGnr_JAVj9PhFnQ,2934
+sm_data_ml_utils/core/config.py,sha256=5i05q4R7VnmmdhxM3owYrJGDH3zTza9vYI5KxxW8Ur4,693
+sm_data_ml_utils/core/databricks_utils.py,sha256=YLq88WzSxfUegwJofEfeTwq6lQslLLDzgA7w1F0IKBk,2935
 sm_data_ml_utils/databricks_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sm_data_ml_utils/databricks_client/client.py,sha256=IbGBDzGVtCsW9NSx2_GL9hXT0VHWsC3_fEzhZvkU9fA,1416
 sm_data_ml_utils/databricks_client/databricks-sql.ipynb,sha256=YFH-TqOB_6dkuBl_qu7wlPmCGy5absciFiQV03FZl9k,41487
 sm_data_ml_utils/mlflow_databricks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sm_data_ml_utils/mlflow_databricks/mlflow_model_utils.py,sha256=974W4Edh9Kyk84BLDmQMkSWCRxbIqi5n36WhqpCBuVY,11866
+sm_data_ml_utils/mlflow_databricks/mlflow_model_utils.py,sha256=0GyBvqZDdZk32h_DiI4DvbVfjOImiHwUAT9ksG-TpIY,12101
 sm_data_ml_utils/mlflow_databricks/mlflow_prediction_requests.py,sha256=H4Z7LVqNim95ljLtYgIjso6rvBj9wqjYTSy8Xog15x8,2578
 sm_data_ml_utils/mlflow_databricks/mlflow_serve.py,sha256=1oM9hngT444lMHf_B_J1xpGaztIzF3rpJ3r5EDLaTZM,6818
 sm_data_ml_utils/mlflow_databricks/mlflow_tracker.py,sha256=TBCDRfEY_yhuet8SPF2U5vNGkGUEzdvEQgkNB_SbMO4,5601
-sm_data_ml_utils-0.1.1.dist-info/LICENSE.txt,sha256=jQGhVAgIuNn_aOSzBgcEjOGhbiYFhdZAmmksZ7Uo85U,1071
-sm_data_ml_utils-0.1.1.dist-info/METADATA,sha256=9jvpPCm4JNoxYl96pq26lVZCHPYg6fnYkZkRAaL5df8,3841
-sm_data_ml_utils-0.1.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-sm_data_ml_utils-0.1.1.dist-info/RECORD,,
+sm_data_ml_utils-0.1.3.dist-info/LICENSE.txt,sha256=jQGhVAgIuNn_aOSzBgcEjOGhbiYFhdZAmmksZ7Uo85U,1071
+sm_data_ml_utils-0.1.3.dist-info/METADATA,sha256=pDvagT6R1BnoxnWneUA9V3ZodnIAODLuzDa4cfrgy1E,3841
+sm_data_ml_utils-0.1.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+sm_data_ml_utils-0.1.3.dist-info/RECORD,,