mlrun 1.10.0rc18__py3-none-any.whl → 1.11.0rc16__py3-none-any.whl

mlrun/db/nopdb.py

@@ -212,7 +212,6 @@ class NopDB(RunDBInterface):
         tree: Optional[str] = None,
         parent: Optional[str] = None,
         format_: mlrun.common.formatters.ArtifactFormat = mlrun.common.formatters.ArtifactFormat.full,
-        limit: Optional[int] = None,
         partition_by: Optional[
             Union[mlrun.common.schemas.ArtifactPartitionByField, str]
         ] = None,
@@ -376,16 +375,6 @@ class NopDB(RunDBInterface):
     ) -> dict:
         pass
 
-    def list_features(
-        self,
-        project: str,
-        name: Optional[str] = None,
-        tag: Optional[str] = None,
-        entities: Optional[list[str]] = None,
-        labels: Optional[Union[str, dict[str, Optional[str]], list[str]]] = None,
-    ) -> mlrun.common.schemas.FeaturesOutput:
-        pass
-
     def list_features_v2(
         self,
         project: str,
@@ -593,16 +582,6 @@ class NopDB(RunDBInterface):
     ):
         pass
 
-    def create_user_secrets(
-        self,
-        user: str,
-        provider: Union[
-            str, mlrun.common.schemas.SecretProviderName
-        ] = mlrun.common.schemas.SecretProviderName.vault,
-        secrets: Optional[dict] = None,
-    ):
-        pass
-
     def create_model_endpoint(
         self,
         model_endpoint: mlrun.common.schemas.ModelEndpoint,
@@ -636,6 +615,9 @@ class NopDB(RunDBInterface):
         tsdb_metrics: bool = False,
         metric_list: Optional[list[str]] = None,
         top_level: bool = False,
+        modes: Optional[
+            Union[mm_constants.EndpointMode, list[mm_constants.EndpointMode]]
+        ] = None,
         uids: Optional[list[str]] = None,
         latest_only: bool = False,
     ) -> mlrun.common.schemas.ModelEndpointList:
@@ -682,6 +664,7 @@ class NopDB(RunDBInterface):
         item_name: Optional[str] = None,
         tag: Optional[str] = None,
         version: Optional[str] = None,
+        item_type: mlrun.common.schemas.hub.HubSourceType = mlrun.common.schemas.hub.HubSourceType.functions,
     ):
         pass
 
@@ -698,6 +681,7 @@ class NopDB(RunDBInterface):
         version: Optional[str] = None,
         tag: Optional[str] = None,
         force_refresh: bool = False,
+        object_type: mlrun.common.schemas.hub.HubSourceType = mlrun.common.schemas.hub.HubSourceType.functions,
     ):
         pass
 
@@ -709,6 +693,18 @@ class NopDB(RunDBInterface):
         version: Optional[str] = None,
         tag: str = "latest",
         force_refresh: bool = False,
+        item_type: mlrun.common.schemas.hub.HubSourceType = mlrun.common.schemas.hub.HubSourceType.functions,
+    ):
+        pass
+
+    def get_hub_asset(
+        self,
+        source_name: str,
+        item_name: str,
+        asset_name: str,
+        version: Optional[str] = None,
+        tag: str = "latest",
+        item_type: mlrun.common.schemas.hub.HubSourceType = mlrun.common.schemas.hub.HubSourceType.functions,
     ):
         pass
 
@@ -895,6 +891,14 @@ class NopDB(RunDBInterface):
     ) -> None:
         pass
 
+    def delete_model_monitoring_metrics(
+        self,
+        project: str,
+        application_name: str,
+        endpoint_ids: Optional[list[str]] = None,
+    ) -> None:
+        pass
+
     def get_monitoring_function_summaries(
         self,
         project: str,
@@ -993,3 +997,32 @@ class NopDB(RunDBInterface):
         end: Optional[datetime.datetime] = None,
     ) -> mlrun.common.schemas.model_monitoring.ModelEndpointDriftValues:
         pass
+
+    def store_secret_token(
+        self,
+        secret_token: mlrun.common.schemas.SecretToken,
+        log_warning: bool = True,
+        force: bool = False,
+    ) -> mlrun.common.schemas.StoreSecretTokensResponse:
+        pass
+
+    def store_secret_tokens(
+        self,
+        secret_tokens: list[mlrun.common.schemas.SecretToken],
+        log_warning: bool = True,
+        force: bool = False,
+    ) -> mlrun.common.schemas.StoreSecretTokensResponse:
+        pass
+
+    def list_secret_tokens(self) -> mlrun.common.schemas.ListSecretTokensResponse:
+        pass
+
+    def revoke_secret_token(self, token_name: str) -> None:
+        pass
+
+    def get_secret_token(
+        self,
+        token_name: str,
+        username: Optional[str] = None,
+    ) -> mlrun.common.schemas.SecretToken:
+        pass

mlrun/errors.py

@@ -264,12 +264,14 @@ class MLRunFatalFailureError(Exception):
 
 
 class ModelRunnerError(MLRunBaseError):
-    def __init__(self, models_errors: dict[str:str], *args) -> None:
+    def __init__(self, models_errors: dict[str, str], *args) -> None:
         self.models_errors = models_errors
         super().__init__(self.__repr__(), *args)
 
     def __repr__(self):
-        return f"ModelRunnerError: {repr(self.models_errors)}"
+        return "ModelRunnerError: " + ";\n".join(
+            f"{model} {msg}" for model, msg in self.models_errors.items()
+        )
 
     def __copy__(self):
         return type(self)(models_errors=self.models_errors)

mlrun/execution.py

@@ -15,7 +15,6 @@
 import logging
 import os
 import uuid
-import warnings
 from copy import deepcopy
 from typing import Optional, Union, cast
 
@@ -101,6 +100,7 @@ class MLClientCtx:
         self._node_selector = {}
         self._tolerations = {}
         self._affinity = {}
+        self._auth = {}
 
         self._function = ""
         self._parameters = {}
@@ -234,6 +234,11 @@ class MLClientCtx:
         """Dictionary with node selectors (read-only)"""
         return deepcopy(self._node_selector)
 
+    @property
+    def auth(self):
+        """Dictionary with auth (read-only)"""
+        return deepcopy(self._auth)
+
     @property
     def tolerations(self):
         """Dictionary with tolerations (read-only)"""
@@ -437,6 +442,7 @@ class MLClientCtx:
             self._affinity = spec.get("affinity", self._affinity)
             self._reset_on_run = spec.get("reset_on_run", self._reset_on_run)
             self._retry_spec = spec.get("retry", self._retry_spec)
+            self._auth = spec.get("auth", self._auth)
 
         self._init_dbs(rundb)
 
@@ -917,7 +923,7 @@ class MLClientCtx:
         prompt_path: Optional[str] = None,
         prompt_legend: Optional[dict] = None,
         model_artifact: Union[ModelArtifact, str] = None,
-        model_configuration: Optional[dict] = None,
+        invocation_config: Optional[dict] = None,
         description: Optional[str] = None,
         target_path: Optional[str] = None,
         artifact_path: Optional[str] = None,
@@ -934,14 +940,51 @@ class MLClientCtx:
 
         Examples::
 
-            # Log an inline prompt
+            # Log directly with an inline prompt template
+            context.log_llm_prompt(
+                key="customer_support_prompt",
+                prompt_template=[
+                    {
+                        "role": "system",
+                        "content": "You are a helpful customer support assistant.",
+                    },
+                    {
+                        "role": "user",
+                        "content": "The customer reports: {issue_description}",
+                    },
+                ],
+                prompt_legend={
+                    "issue_description": {
+                        "field": "user_issue",
+                        "description": "Detailed description of the customer's issue",
+                    },
+                    "solution": {
+                        "field": "proposed_solution",
+                        "description": "Suggested fix for the customer's issue",
+                    },
+                },
+                model_artifact=model,
+                invocation_config={"temperature": 0.5, "max_tokens": 200},
+                description="Prompt for handling customer support queries",
+                tag="support-v1",
+                labels={"domain": "support"},
+            )
+
+            # Log a prompt from file
             context.log_llm_prompt(
-                key="qa-prompt",
-                prompt_template=[{"role: "user", "content": "question with {place_holder}"}],
+                key="qa_prompt",
+                prompt_path="prompts/template.json",
+                prompt_legend={
+                    "question": {
+                        "field": "user_question",
+                        "description": "The actual question asked by the user",
+                    }
+                },
                 model_artifact=model,
-                prompt_legend={"question": "user_input"},
-                model_configuration={"temperature": 0.7, "max_tokens": 128},
-                tag="latest",
+                invocation_config={"temperature": 0.7, "max_tokens": 256},
+                description="Q&A prompt template with user-provided question",
+                tag="v2",
+                labels={"task": "qa", "stage": "experiment"},
             )
 
         :param key: Unique name of the artifact.
@@ -950,17 +993,22 @@ class MLClientCtx:
          "role": "user", "content": "I need your help with {profession}"]. only "role" and "content" keys allow in any
          str format (upper/lower case), keys will be modified to lower case.
          Cannot be used with `prompt_path`.
-        :param prompt_path: Path to a file containing the prompt content. Cannot be used with `prompt_string`.
+        :param prompt_path: Path to a JSON file containing the prompt template.
+                    Cannot be used together with `prompt_template`.
+                    The file should define a list of dictionaries in the same format
+                    supported by `prompt_template`.
         :param prompt_legend: A dictionary where each key is a placeholder in the prompt (e.g., ``{user_name}``)
                and the value is a dictionary holding two keys, "field", "description". "field" points to the field in
                the event where the value of the place-holder inside the event, if None or not exist will be replaced
                with the place-holder name. "description" will point to explanation of what that placeholder represents.
                Useful for documenting and clarifying dynamic parts of the prompt.
         :param model_artifact: Reference to the parent model (either `ModelArtifact` or model URI string).
-        :param model_configuration: Dictionary of generation parameters (e.g., temperature, max_tokens).
-        :param description: Optional description of the prompt.
-        :param target_path: Path to write the artifact locally.
-        :param artifact_path: Path in the artifact store (defaults to project artifact path).
+        :param invocation_config: Dictionary of generation parameters (e.g., temperature, max_tokens).
+        :param description:   Optional description of the prompt.
+        :param target_path:   Absolute target path (instead of using artifact_path + local_path)
+        :param artifact_path: Target artifact path (when not using the default)
+                              To define a subpath under the default location use:
+                              `artifact_path=context.artifact_subpath('data')`
         :param tag: Tag/version to assign to the prompt artifact.
         :param labels: Labels to tag the artifact (e.g., list or dict of key-value pairs).
         :param upload: Whether to upload the artifact to the store (defaults to True).
@@ -981,7 +1029,7 @@ class MLClientCtx:
             prompt_path=prompt_path,
             prompt_legend=prompt_legend,
             model_artifact=model_artifact,
-            model_configuration=model_configuration,
+            invocation_config=invocation_config,
             target_path=target_path,
             description=description,
             **kwargs,
@@ -1099,14 +1147,6 @@ class MLClientCtx:
         self._update_run()
         return item
 
-    def get_cached_artifact(self, key):
-        """Return a logged artifact from cache (for potential updates)"""
-        warnings.warn(
-            "get_cached_artifact is deprecated in 1.8.0 and will be removed in 1.11.0. Use get_artifact instead.",
-            FutureWarning,
-        )
-        return self.get_artifact(key)
-
     def get_artifact(
         self, key, tag=None, iter=None, tree=None, uid=None
     ) -> Optional[Artifact]:
@@ -1268,6 +1308,7 @@ class MLClientCtx:
                 "node_selector": self._node_selector,
                 "tolerations": self._tolerations,
                 "affinity": self._affinity,
+                "auth": self._auth,
                 "retry": self._retry_spec,
             },
             "status": {
@@ -1469,15 +1510,15 @@ class MLClientCtx:
 
 
 def _cast_result(value):
-    if isinstance(value, (int, str, float)):
+    if isinstance(value, int | str | float):
         return value
     if isinstance(value, list):
         return [_cast_result(v) for v in value]
     if isinstance(value, dict):
         return {k: _cast_result(v) for k, v in value.items()}
-    if isinstance(value, (np.int64, np.integer)):
+    if isinstance(value, np.int64 | np.integer):
         return int(value)
-    if isinstance(value, (np.floating, np.float64)):
+    if isinstance(value, np.floating | np.float64):
         return float(value)
     if isinstance(value, np.ndarray):
         return value.tolist()

mlrun/feature_store/api.py

@@ -361,7 +361,7 @@ def _ingest(
         import pyspark.sql
 
         if (
-            isinstance(source, (pd.DataFrame, pyspark.sql.DataFrame))
+            isinstance(source, pd.DataFrame | pyspark.sql.DataFrame)
             and run_config is not None
         ):
             raise mlrun.errors.MLRunInvalidArgumentError(

mlrun/feature_store/common.py

@@ -256,7 +256,7 @@ class RunConfig:
     @function.setter
     def function(self, function):
         if function and not (
-            isinstance(function, (str, FunctionReference)) or hasattr(function, "apply")
+            isinstance(function, str | FunctionReference) or hasattr(function, "apply")
         ):
             raise mlrun.errors.MLRunInvalidArgumentError(
                 "function must be a uri (string) or mlrun function object/reference"

mlrun/feature_store/feature_vector_utils.py

@@ -370,7 +370,7 @@ class OnlineVectorService:
         if (
             not entity_rows
             or not isinstance(entity_rows, list)
-            or not isinstance(entity_rows[0], (list, dict))
+            or not isinstance(entity_rows[0], list | dict)
         ):
             raise mlrun.errors.MLRunInvalidArgumentError(
                 f"input data is of type {type(entity_rows)}. must be a list of lists or list of dicts"

mlrun/feature_store/steps.py

@@ -279,11 +279,11 @@ class MapValues(StepToDict, MLRunStep):
                     new_col_type = df.schema[new_column_name].dataType
                     #  in order to avoid exception at isna on non-decimal/float columns -
                     #  we need to check their types before filtering.
-                    if isinstance(col_type, (FloatType, DoubleType, DecimalType)):
+                    if isinstance(col_type, FloatType | DoubleType | DecimalType):
                         column_filter = (~isnull(col(column))) & (~isnan(col(column)))
                     else:
                         column_filter = ~isnull(col(column))
-                    if isinstance(new_col_type, (FloatType, DoubleType, DecimalType)):
+                    if isinstance(new_col_type, FloatType | DoubleType | DecimalType):
                         new_column_filter = isnull(col(new_column_name)) | isnan(
                             col(new_column_name)
                         )
@@ -295,7 +295,7 @@ class MapValues(StepToDict, MLRunStep):
                         for k, v in column_map.items()
                         if v is None
                         or (
-                            isinstance(v, (float, np.float64, np.float32, np.float16))
+                            isinstance(v, float | np.float64 | np.float32 | np.float16)
                             and math.isnan(v)
                         )
                     ]
@@ -338,7 +338,7 @@ class MapValues(StepToDict, MLRunStep):
                     for val in column_map.values()
                     if type(val) is not None
                     and not (
-                        isinstance(val, (float, np.float64, np.float32, np.float16))
+                        isinstance(val, float | np.float64 | np.float32 | np.float16)
                         and math.isnan(val)
                     )
                 )
@@ -358,7 +358,9 @@ class MapValues(StepToDict, MLRunStep):
                         and val != "-inf"
                         and val != "inf"
                         and not (
-                            isinstance(val, (float, np.float64, np.float32, np.float16))
+                            isinstance(
+                                val, float | np.float64 | np.float32 | np.float16
+                            )
                             and math.isnan(val)
                         )
                     )
@@ -443,7 +445,7 @@ class OneHotEncoder(StepToDict, MLRunStep):
         self.mapping = mapping
         for key, values in mapping.items():
             for val in values:
-                if not (isinstance(val, str) or isinstance(val, (int, np.integer))):
+                if not (isinstance(val, str) or isinstance(val, int | np.integer)):
                     raise mlrun.errors.MLRunInvalidArgumentError(
                         "For OneHotEncoder you must provide int or string mapping list"
                     )

mlrun/frameworks/_common/utils.py

@@ -97,9 +97,9 @@ class CommonUtils(ABC):
         """
         if isinstance(dataset, np.ndarray):
             return dataset
-        if isinstance(dataset, (pd.DataFrame, pd.Series)):
+        if isinstance(dataset, pd.DataFrame | pd.Series):
             return dataset.to_numpy()
-        if isinstance(dataset, (list, tuple)):
+        if isinstance(dataset, list | tuple):
             return np.array(dataset)
         if isinstance(dataset, dict):
             return np.array(list(dataset.values()))
@@ -133,7 +133,7 @@ class CommonUtils(ABC):
         """
         if isinstance(dataset, pd.DataFrame):
             return dataset
-        if isinstance(dataset, (np.ndarray, pd.Series, list, tuple, dict)):
+        if isinstance(dataset, np.ndarray | pd.Series | list | tuple | dict):
             return pd.DataFrame(dataset)
         try:
             # SciPy is not in MLRun's requirements but common to all frameworks.

mlrun/frameworks/_dl_common/loggers/logger.py

@@ -281,7 +281,7 @@ class Logger:
         """
         for parameter_name, parameter_value in self._context.parameters.items():
             # Check if the parameter is a trackable value:
-            if isinstance(parameter_value, (str, bool, float, int)):
+            if isinstance(parameter_value, str | bool | float | int):
                 self.log_static_hyperparameter(
                     parameter_name=parameter_name, value=parameter_value
                 )

mlrun/frameworks/_dl_common/loggers/tensorboard_logger.py

@@ -14,8 +14,9 @@
 
 import os
 from abc import abstractmethod
+from collections.abc import Callable
 from datetime import datetime
-from typing import Any, Callable, Generic, Optional, Union
+from typing import Any, Generic, Optional, Union
 
 import yaml
 

mlrun/frameworks/_ml_common/loggers/mlrun_logger.py

@@ -73,7 +73,7 @@ class MLRunLogger(Logger):
         """
         for parameter_name, parameter_value in self._context.parameters.items():
             # Check if the parameter is a trackable value:
-            if isinstance(parameter_value, (str, bool, float, int)):
+            if isinstance(parameter_value, str | bool | float | int):
                 self.log_static_hyperparameter(
                     parameter_name=parameter_name, value=parameter_value
                 )

mlrun/frameworks/_ml_common/utils.py

@@ -13,8 +13,9 @@
 # limitations under the License.
 
 from abc import ABC
+from collections.abc import Callable
 from enum import Enum
-from typing import Callable, Union
+from typing import Union
 
 import pandas as pd
 from sklearn.base import is_classifier, is_regressor

mlrun/frameworks/auto_mlrun/auto_mlrun.py

@@ -12,7 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Callable, Optional, Union
+from collections.abc import Callable
+from typing import Optional, Union
 
 import mlrun
 from mlrun.artifacts import get_model
@@ -58,7 +59,7 @@ def get_framework_by_instance(model: CommonTypes.ModelType) -> str:
 
         from mlrun.frameworks.xgboost import XGBoostModelHandler
 
-        if isinstance(model, (XGBModel, Booster)):
+        if isinstance(model, XGBModel | Booster):
             return XGBoostModelHandler.FRAMEWORK_NAME
     except ModuleNotFoundError:
         pass
@@ -69,7 +70,7 @@ def get_framework_by_instance(model: CommonTypes.ModelType) -> str:
 
         from mlrun.frameworks.lgbm import LGBMModelHandler
 
-        if isinstance(model, (LGBMModel, Booster)):
+        if isinstance(model, LGBMModel | Booster):
             return LGBMModelHandler.FRAMEWORK_NAME
     except ModuleNotFoundError:
         pass

mlrun/frameworks/lgbm/mlrun_interfaces/mlrun_interface.py

@@ -13,8 +13,9 @@
 # limitations under the License.
 
 from abc import ABC
+from collections.abc import Callable
 from types import ModuleType
-from typing import Callable, Optional, Union
+from typing import Optional, Union
 
 import lightgbm as lgb
 

mlrun/frameworks/onnx/dataset.py

@@ -13,7 +13,8 @@
 # limitations under the License.
 
 import math
-from typing import Callable, Optional, Union
+from collections.abc import Callable
+from typing import Optional, Union
 
 import numpy as np
 

mlrun/frameworks/onnx/mlrun_interface.py

@@ -12,7 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Callable, Optional
+from collections.abc import Callable
+from typing import Optional
 
 import numpy as np
 import onnx

mlrun/frameworks/pytorch/callbacks/logging_callback.py

@@ -12,7 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Callable, Optional, Union
+from collections.abc import Callable
+from typing import Optional, Union
 
 import numpy as np
 from torch import Tensor
@@ -496,7 +497,7 @@ class LoggingCallback(Callback):
             value = self._objects[source]
             for key in key_chain:
                 try:
-                    if isinstance(value, (dict, list, tuple)):
+                    if isinstance(value, dict | list | tuple):
                         value = value[key]
                     else:
                         value = getattr(value, key)
@@ -507,7 +508,7 @@ class LoggingCallback(Callback):
                     )
 
         # Parse the value:
-        if isinstance(value, (Tensor, Parameter)):
+        if isinstance(value, Tensor | Parameter):
             if value.numel() == 1:
                 value = float(value)
             else:
@@ -523,7 +524,7 @@ class LoggingCallback(Callback):
                     f"The parameter with the following key chain: {key_chain} is a numpy.ndarray with {value.size} "
                     f"elements. numpy arrays are trackable only if they have 1 element."
                 )
-        elif not isinstance(value, (float, int, str, bool)):
+        elif not isinstance(value, float | int | str | bool):
             raise mlrun.errors.MLRunInvalidArgumentError(
                 f"The parameter with the following key chain: {key_chain} is of type '{type(value)}'. "
                 f"The only trackable types are: float, int, str and bool."

mlrun/frameworks/pytorch/callbacks/mlrun_logging_callback.py

@@ -12,7 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Callable, Optional, Union
+from collections.abc import Callable
+from typing import Optional, Union
 
 import torch
 from torch import Tensor

mlrun/frameworks/pytorch/callbacks/tensorboard_logging_callback.py

@@ -12,8 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from collections.abc import Callable
 from datetime import datetime
-from typing import Callable, Optional, Union
+from typing import Optional, Union
 
 import torch
 from torch import Tensor

mlrun/frameworks/pytorch/utils.py

@@ -12,7 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Callable, Union
+from collections.abc import Callable
+from typing import Union
 
 import numpy as np
 import torch

mlrun/frameworks/sklearn/metric.py

@@ -15,7 +15,8 @@
 import importlib
 import json
 import sys
-from typing import Callable, Optional, Union
+from collections.abc import Callable
+from typing import Optional, Union
 
 import mlrun.errors
 

mlrun/frameworks/tf_keras/callbacks/logging_callback.py

@@ -12,7 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Callable, Optional, Union
+from collections.abc import Callable
+from typing import Optional, Union
 
 import numpy as np
 import tensorflow as tf
@@ -441,8 +442,8 @@ class LoggingCallback(Callback):
                     )
 
         # Parse the value:
-        if isinstance(value, (tf.Tensor, tf.Variable)) or (
-            is_keras_3() and isinstance(value, (keras.KerasTensor, keras.Variable))
+        if isinstance(value, tf.Tensor | tf.Variable) or (
+            is_keras_3() and isinstance(value, keras.KerasTensor | keras.Variable)
         ):
             if int(tf.size(value)) == 1:
                 value = float(value)
@@ -459,7 +460,7 @@ class LoggingCallback(Callback):
                     f"The parameter with the following key chain: {key_chain} is a numpy.ndarray with {value.size} "
                     f"elements. numpy arrays are trackable only if they have 1 element."
                 )
-        elif not (isinstance(value, (float, int, str, bool))):
+        elif not (isinstance(value, float | int | str | bool)):
             raise mlrun.errors.MLRunInvalidArgumentError(
                 f"The parameter with the following key chain: {key_chain} is of type '{type(value)}'. The only "
                 f"trackable types are: float, int, str and bool."

mlrun/frameworks/tf_keras/callbacks/mlrun_logging_callback.py

@@ -12,7 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Callable, Optional, Union
+from collections.abc import Callable
+from typing import Optional, Union
 
 import mlrun
 from mlrun.artifacts import Artifact