mlrun 1.6.4rc7__py3-none-any.whl → 1.7.0__py3-none-any.whl

mlrun/frameworks/_dl_common/loggers/tensorboard_logger.py

@@ -15,7 +15,7 @@
 import os
 from abc import abstractmethod
 from datetime import datetime
-from typing import Any, Callable, Dict, Generic, List, Union
+from typing import Any, Callable, Generic, Union
 
 import yaml
 
@@ -60,7 +60,7 @@ class TensorboardLogger(Logger, Generic[DLTypes.WeightType]):
 
     def __init__(
         self,
-        statistics_functions: List[
+        statistics_functions: list[
             Callable[[DLTypes.WeightType], Union[float, DLTypes.WeightType]]
         ],
         context: mlrun.MLClientCtx = None,
@@ -94,7 +94,7 @@ class TensorboardLogger(Logger, Generic[DLTypes.WeightType]):
         :raise MLRunInvalidArgumentError: If the `update_frequency` is illegal or if `tensorboard_directory` and
                                           `context` were not given.
         """
-        super(TensorboardLogger, self).__init__(context=context)
+        super().__init__(context=context)
 
         # Validate the context and tensorboard directory combination:
         if tensorboard_directory is None and context is None:
@@ -136,7 +136,7 @@ class TensorboardLogger(Logger, Generic[DLTypes.WeightType]):
             self._weights_statistics[statistic_function.__name__] = {}  # type: Dict[str, List[float]]
 
     @property
-    def weights(self) -> Dict[str, DLTypes.WeightType]:
+    def weights(self) -> dict[str, DLTypes.WeightType]:
         """
         Get the logged weights dictionary. Each of the logged weight will be found by its name.
 
@@ -145,7 +145,7 @@ class TensorboardLogger(Logger, Generic[DLTypes.WeightType]):
         return self._weights
 
     @property
-    def weight_statistics(self) -> Dict[str, Dict[str, List[float]]]:
+    def weight_statistics(self) -> dict[str, dict[str, list[float]]]:
         """
         Get the logged statistics for all the tracked weights. Each statistic has a dictionary of weights and their list
         of epochs values.
@@ -514,23 +514,15 @@ class TensorboardLogger(Logger, Generic[DLTypes.WeightType]):
                 + list(self._dynamic_hyperparameters.keys()),
             ],
         ):
-            text += "\n  * **{}**: {}".format(
-                property_name.capitalize(),
-                self._markdown_print(value=property_value, tabs=2),
-            )
+            text += f"\n  * **{property_name.capitalize()}**: {self._markdown_print(value=property_value, tabs=2)}"
 
         # Add the context state:
         if self._context is not None:
-            text += "\n####Context initial state: ({})".format(
-                self._generate_context_link(context=self._context)
-            )
+            text += f"\n####Context initial state: ({self._generate_context_link(context=self._context)})"
             for property_name, property_value in self._extract_properties_from_context(
                 context=self._context
             ).items():
-                text += "\n  * **{}**: {}".format(
-                    property_name.capitalize(),
-                    self._markdown_print(value=property_value, tabs=2),
-                )
+                text += f"\n  * **{property_name.capitalize()}**: {self._markdown_print(value=property_value, tabs=2)}"
 
         return text
 
@@ -541,7 +533,7 @@ class TensorboardLogger(Logger, Generic[DLTypes.WeightType]):
 
         :return: The generated text.
         """
-        text = "####Epoch {} summary:".format(self._epochs)
+        text = f"####Epoch {self._epochs} summary:"
         if self._context is not None:
             for property_name, property_value in self._extract_properties_from_context(
                 context=self._context
@@ -555,16 +547,13 @@ class TensorboardLogger(Logger, Generic[DLTypes.WeightType]):
                     "inputs",
                     "parameters",
                 ]:
-                    text += "\n  * **{}**: {}".format(
-                        property_name.capitalize(),
-                        self._markdown_print(value=property_value, tabs=2),
+                    text += (
+                        f"\n  * **{property_name.capitalize()}**: "
+                        f"{self._markdown_print(value=property_value, tabs=2)}"
                     )
         else:
             for property_name, property_value in self._extract_epoch_results().items():
-                text += "\n  * **{}**: {}".format(
-                    property_name.capitalize(),
-                    self._markdown_print(value=property_value, tabs=2),
-                )
+                text += f"\n  * **{property_name.capitalize()}**: {self._markdown_print(value=property_value, tabs=2)}"
         return text
 
     def _generate_run_end_text(self) -> str:
@@ -575,30 +564,22 @@ class TensorboardLogger(Logger, Generic[DLTypes.WeightType]):
         :return: The generated text.
         """
         # Write the run summary:
-        text = "\n####Run final summary - epoch {}:".format(self._epochs)
+        text = f"\n####Run final summary - epoch {self._epochs}:"
         for property_name, property_value in self._extract_epoch_results().items():
-            text += "\n  * **{}**: {}".format(
-                property_name.capitalize(),
-                self._markdown_print(value=property_value, tabs=2),
-            )
+            text += f"\n  * **{property_name.capitalize()}**: {self._markdown_print(value=property_value, tabs=2)}"
 
         # Add the context final state:
         if self._context is not None:
-            text += "\n####Context final state: ({})".format(
-                self._generate_context_link(context=self._context)
-            )
+            text += f"\n####Context final state: ({self._generate_context_link(context=self._context)})"
             for property_name, property_value in self._extract_properties_from_context(
                 context=self._context
             ).items():
-                text += "\n  * **{}**: {}".format(
-                    property_name.capitalize(),
-                    self._markdown_print(value=property_value, tabs=2),
-                )
+                text += f"\n  * **{property_name.capitalize()}**: {self._markdown_print(value=property_value, tabs=2)}"
         return text
 
     def _extract_epoch_results(
         self, epoch: int = -1
-    ) -> Dict[str, Dict[str, DLTypes.TrackableType]]:
+    ) -> dict[str, dict[str, DLTypes.TrackableType]]:
         """
         Extract the given epoch results from all the collected values and results.
 
@@ -633,17 +614,12 @@ class TensorboardLogger(Logger, Generic[DLTypes.WeightType]):
         :return: The generated link.
         """
         return (
-            '<a href="{}/{}/{}/jobs/monitor/{}/overview" target="_blank">{}</a>'.format(
-                config.resolve_ui_url(),
-                config.ui.projects_prefix,
-                context.project,
-                context.uid,
-                link_text,
-            )
+            f'<a href="{config.resolve_ui_url()}/{config.ui.projects_prefix}/{context.project}'
+            f'/jobs/monitor/{context.uid}/overview" target="_blank">{link_text}</a>'
         )
 
     @staticmethod
-    def _extract_properties_from_context(context: mlrun.MLClientCtx) -> Dict[str, Any]:
+    def _extract_properties_from_context(context: mlrun.MLClientCtx) -> dict[str, Any]:
         """
         Extract the properties of the run this context belongs to.
 
@@ -672,13 +648,13 @@ class TensorboardLogger(Logger, Generic[DLTypes.WeightType]):
         if isinstance(value, list):
             if len(value) == 0:
                 return ""
-            text = "\n" + yaml.dump(value)
+            text = "\n" + yaml.safe_dump(value)
             text = "  \n".join(["  " * tabs + line for line in text.splitlines()])
             return text
         if isinstance(value, dict):
             if len(value) == 0:
                 return ""
-            text = yaml.dump(value)
+            text = yaml.safe_dump(value)
             text = "  \n".join(
                 ["  " * tabs + "- " + line for line in text.splitlines()]
             )

mlrun/frameworks/_ml_common/artifacts_library.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 #
 from abc import ABC
-from typing import List
 
 from .._common.artifacts_library import ArtifactsLibrary, Plan
 from .plans import (
@@ -47,7 +46,7 @@ class MLArtifactsLibrary(ArtifactsLibrary, ABC):
     @classmethod
     def default(
         cls, model: MLTypes.ModelType, y: MLTypes.DatasetType = None, *args, **kwargs
-    ) -> List[Plan]:
+    ) -> list[Plan]:
         """
         Get the default artifacts plans list of this framework's library.
 

mlrun/frameworks/_ml_common/loggers/logger.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from typing import Dict, List
 
 from ..utils import MLTypes
 
@@ -42,7 +41,7 @@ class Logger:
         self._iterations = 0
 
     @property
-    def results(self) -> Dict[str, Dict[str, List[float]]]:
+    def results(self) -> dict[str, dict[str, list[float]]]:
         """
         Get the results logged. The results will be stored in a dictionary where each key is the validation set name
         and the value is a dictionary of metrics to their list of iterations values.
@@ -52,7 +51,7 @@ class Logger:
         return self._results
 
     @property
-    def static_hyperparameters(self) -> Dict[str, MLTypes.TrackableType]:
+    def static_hyperparameters(self) -> dict[str, MLTypes.TrackableType]:
         """
         Get the static hyperparameters logged. The hyperparameters will be stored in a dictionary where each key is the
         hyperparameter name and the value is his logged value.
@@ -62,7 +61,7 @@ class Logger:
         return self._static_hyperparameters
 
     @property
-    def dynamic_hyperparameters(self) -> Dict[str, List[MLTypes.TrackableType]]:
+    def dynamic_hyperparameters(self) -> dict[str, list[MLTypes.TrackableType]]:
         """
         Get the dynamic hyperparameters logged. The hyperparameters will be stored in a dictionary where each key is the
         hyperparameter name and the value is a list of his logged values per epoch.

mlrun/frameworks/_ml_common/loggers/mlrun_logger.py

@@ -13,7 +13,6 @@
 # limitations under the License.
 #
 import re
-from typing import Dict, List
 
 import numpy as np
 import plotly.graph_objects as go
@@ -39,7 +38,7 @@ class MLRunLogger(Logger):
 
         :param context: MLRun context to log to. The context parameters can be logged as static hyperparameters.
         """
-        super(MLRunLogger, self).__init__()
+        super().__init__()
 
         # An MLRun context to log to:
         self._context = context
@@ -47,7 +46,7 @@ class MLRunLogger(Logger):
         # Prepare the artifacts dictionary:
         self._artifacts = {}  # type: Dict[str, Artifact]
 
-    def get_artifacts(self) -> Dict[str, Artifact]:
+    def get_artifacts(self) -> dict[str, Artifact]:
         """
         Get the artifacts created by this logger.
 
@@ -55,7 +54,7 @@ class MLRunLogger(Logger):
         """
         return self._artifacts
 
-    def get_metrics(self) -> Dict[str, float]:
+    def get_metrics(self) -> dict[str, float]:
         """
         Generate a metrics summary to log along the model.
 
@@ -144,7 +143,7 @@ class MLRunLogger(Logger):
 
     @staticmethod
     def _produce_convergence_plot_artifact(
-        name: str, values: List[float]
+        name: str, values: list[float]
     ) -> PlotlyArtifact:
         """
         Produce the convergences for the provided metric according.

mlrun/frameworks/_ml_common/model_handler.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 #
 from abc import ABC
-from typing import Dict, List, Union
+from typing import Union
 
 import mlrun
 from mlrun.artifacts import Artifact
@@ -35,10 +35,10 @@ class MLModelHandler(ModelHandler, ABC):
         model_path: MLTypes.PathType = None,
         model_name: str = None,
         modules_map: Union[
-            Dict[str, Union[None, str, List[str]]], MLTypes.PathType
+            dict[str, Union[None, str, list[str]]], MLTypes.PathType
         ] = None,
         custom_objects_map: Union[
-            Dict[str, Union[str, List[str]]], MLTypes.PathType
+            dict[str, Union[str, list[str]]], MLTypes.PathType
         ] = None,
         custom_objects_directory: MLTypes.PathType = None,
         context: mlrun.MLClientCtx = None,
@@ -105,7 +105,7 @@ class MLModelHandler(ModelHandler, ABC):
         self._feature_weights = None  # type: List[float]
 
         # Continue the initialization:
-        super(MLModelHandler, self).__init__(
+        super().__init__(
             model=model,
             model_path=model_path,
             model_name=model_name,
@@ -153,7 +153,7 @@ class MLModelHandler(ModelHandler, ABC):
         return self._feature_vector
 
     @property
-    def feature_weights(self) -> List[float]:
+    def feature_weights(self) -> list[float]:
         """
         Get the feature weights set in this handler.
 
@@ -213,7 +213,7 @@ class MLModelHandler(ModelHandler, ABC):
         """
         self._feature_vector = feature_vector
 
-    def set_feature_weights(self, feature_weights: List[float]):
+    def set_feature_weights(self, feature_weights: list[float]):
         """
         Set the feature weights this model will be logged with.
 
@@ -224,18 +224,18 @@ class MLModelHandler(ModelHandler, ABC):
     def log(
         self,
         tag: str = "",
-        labels: Dict[str, Union[str, int, float]] = None,
-        parameters: Dict[str, Union[str, int, float]] = None,
-        inputs: List[Feature] = None,
-        outputs: List[Feature] = None,
-        metrics: Dict[str, Union[int, float]] = None,
-        artifacts: Dict[str, Artifact] = None,
-        extra_data: Dict[str, MLTypes.ExtraDataType] = None,
+        labels: dict[str, Union[str, int, float]] = None,
+        parameters: dict[str, Union[str, int, float]] = None,
+        inputs: list[Feature] = None,
+        outputs: list[Feature] = None,
+        metrics: dict[str, Union[int, float]] = None,
+        artifacts: dict[str, Artifact] = None,
+        extra_data: dict[str, MLTypes.ExtraDataType] = None,
         algorithm: str = None,
         sample_set: MLTypes.DatasetType = None,
         target_columns: MLTypes.TargetColumnsNamesType = None,
         feature_vector: str = None,
-        feature_weights: List[float] = None,
+        feature_weights: list[float] = None,
     ):
         """
         Log the model held by this handler into the MLRun context provided.
@@ -281,7 +281,7 @@ class MLModelHandler(ModelHandler, ABC):
             self.set_feature_weights(feature_weights=feature_weights)
 
         # Continue with the handler logging:
-        super(MLModelHandler, self).log(
+        super().log(
             tag=tag,
             labels=labels,
             parameters=parameters,
@@ -299,15 +299,15 @@ class MLModelHandler(ModelHandler, ABC):
 
     def update(
         self,
-        labels: Dict[str, Union[str, int, float]] = None,
-        parameters: Dict[str, Union[str, int, float]] = None,
-        inputs: List[Feature] = None,
-        outputs: List[Feature] = None,
-        metrics: Dict[str, Union[int, float]] = None,
-        artifacts: Dict[str, Artifact] = None,
-        extra_data: Dict[str, MLTypes.ExtraDataType] = None,
+        labels: dict[str, Union[str, int, float]] = None,
+        parameters: dict[str, Union[str, int, float]] = None,
+        inputs: list[Feature] = None,
+        outputs: list[Feature] = None,
+        metrics: dict[str, Union[int, float]] = None,
+        artifacts: dict[str, Artifact] = None,
+        extra_data: dict[str, MLTypes.ExtraDataType] = None,
         feature_vector: str = None,
-        feature_weights: List[float] = None,
+        feature_weights: list[float] = None,
     ):
         """
         Update the model held by this handler into the MLRun context provided, updating the model's artifact properties
@@ -336,7 +336,7 @@ class MLModelHandler(ModelHandler, ABC):
             self._feature_weights = feature_weights
 
         # Continue with the handler update:
-        super(MLModelHandler, self).update(
+        super().update(
             labels=labels,
             parameters=parameters,
             inputs=inputs,

mlrun/frameworks/_ml_common/pkl_model_server.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from typing import Any, Dict
+from typing import Any
 
 import numpy as np
 import pandas as pd
@@ -59,7 +59,7 @@ class PickleModelServer(V2ModelServer):
 
         return y_pred.tolist()
 
-    def explain(self, request: Dict[str, Any]) -> str:
+    def explain(self, request: dict[str, Any]) -> str:
         """
         Returns a string listing the model that is being served in this serving function and the function name.
 

mlrun/frameworks/_ml_common/plan.py

@@ -16,7 +16,7 @@ import json
 from abc import ABC, abstractmethod
 from enum import Enum
 
-from IPython.core.display import HTML, display
+from IPython.display import HTML, display
 
 import mlrun
 
@@ -57,7 +57,7 @@ class MLPlan(Plan, ABC):
                                    False.
         """
         self._need_probabilities = need_probabilities
-        super(MLPlan, self).__init__()
+        super().__init__()
 
     @property
     def need_probabilities(self) -> bool:

mlrun/frameworks/_ml_common/plans/calibration_curve_plan.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from typing import Dict
 
 import plotly.graph_objects as go
 from sklearn.calibration import calibration_curve
@@ -51,7 +50,7 @@ class CalibrationCurvePlan(MLPlotPlan):
         self._strategy = strategy
 
         # Continue the initialization for the MLPlan:
-        super(CalibrationCurvePlan, self).__init__(need_probabilities=True)
+        super().__init__(need_probabilities=True)
 
     def is_ready(self, stage: MLPlanStages, is_probabilities: bool) -> bool:
         """
@@ -73,7 +72,7 @@ class CalibrationCurvePlan(MLPlotPlan):
         model: MLTypes.ModelType = None,
         x: MLTypes.DatasetType = None,
         **kwargs,
-    ) -> Dict[str, Artifact]:
+    ) -> dict[str, Artifact]:
         """
         Produce the calibration curve according to the ground truth (y) and predictions (y_pred) values. If predictions
         are not available, the model and a dataset can be given to produce them.

mlrun/frameworks/_ml_common/plans/confusion_matrix_plan.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from typing import Dict
 
 import numpy as np
 import pandas as pd
@@ -57,7 +56,7 @@ class ConfusionMatrixPlan(MLPlotPlan):
         self._normalize = normalize
 
         # Continue the initialization for the MLPlan:
-        super(ConfusionMatrixPlan, self).__init__()
+        super().__init__()
 
     def is_ready(self, stage: MLPlanStages, is_probabilities: bool) -> bool:
         """
@@ -79,7 +78,7 @@ class ConfusionMatrixPlan(MLPlotPlan):
         model: MLTypes.ModelType = None,
         x: MLTypes.DatasetType = None,
         **kwargs,
-    ) -> Dict[str, Artifact]:
+    ) -> dict[str, Artifact]:
         """
         Produce the confusion matrix according to the ground truth (y) and predictions (y_pred) values. If predictions
         are not available, the model and a dataset can be given to produce them.

mlrun/frameworks/_ml_common/plans/dataset_plan.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 #
 from enum import Enum
-from typing import Dict, Union
+from typing import Union
 
 import mlrun.errors
 from mlrun.artifacts import Artifact, DatasetArtifact
@@ -92,7 +92,7 @@ class DatasetPlan(MLPlan):
         self._plans = {}  # TODO: Implement DatasetPlansLibrary with dataset specific artifacts plans.
 
         # Continue initializing the plan:
-        super(DatasetPlan, self).__init__(need_probabilities=False)
+        super().__init__(need_probabilities=False)
 
     def is_ready(self, stage: MLPlanStages, is_probabilities: bool) -> bool:
         """
@@ -124,7 +124,7 @@ class DatasetPlan(MLPlan):
         y: MLTypes.DatasetType = None,
         target_columns_names: MLTypes.TargetColumnsNamesType = None,
         **kwargs,
-    ) -> Dict[str, Artifact]:
+    ) -> dict[str, Artifact]:
         """
         Produce the dataset artifact according to this plan.
 

mlrun/frameworks/_ml_common/plans/feature_importance_plan.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from typing import Dict, Union
+from typing import Union
 
 import numpy as np
 import plotly.graph_objects as go
@@ -38,7 +38,7 @@ class FeatureImportancePlan(MLPlotPlan):
         An example of use can be seen at the Scikit-Learn docs here:
         https://scikit-learn.org/stable/auto_examples/ensemble/plot_forest_importances.html
         """
-        super(FeatureImportancePlan, self).__init__()
+        super().__init__()
 
     def is_ready(self, stage: MLPlanStages, is_probabilities: bool) -> bool:
         """
@@ -55,7 +55,7 @@ class FeatureImportancePlan(MLPlotPlan):
 
     def produce(
         self, model: MLTypes.ModelType, x: MLTypes.DatasetType, **kwargs
-    ) -> Dict[str, Artifact]:
+    ) -> dict[str, Artifact]:
         """
         Produce the feature importance according to the given model and dataset ('x').
 

mlrun/frameworks/_ml_common/plans/roc_curve_plan.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-from typing import Dict, List, Union
+from typing import Union
 
 import numpy as np
 import pandas as pd
@@ -41,7 +41,7 @@ class ROCCurvePlan(MLPlotPlan):
         average: str = "macro",
         max_fpr: float = None,
         multi_class: str = "raise",
-        labels: List[str] = None,
+        labels: list[str] = None,
     ):
         """
         Initialize a receiver operating characteristic plan with the given configuration.
@@ -75,7 +75,7 @@ class ROCCurvePlan(MLPlotPlan):
         self._labels = labels
 
         # Continue the initialization for the MLPlan:
-        super(ROCCurvePlan, self).__init__(need_probabilities=True)
+        super().__init__(need_probabilities=True)
 
     def is_ready(self, stage: MLPlanStages, is_probabilities: bool) -> bool:
         """
@@ -97,7 +97,7 @@ class ROCCurvePlan(MLPlotPlan):
         model: MLTypes.ModelType = None,
         x: MLTypes.DatasetType = None,
         **kwargs,
-    ) -> Dict[str, Artifact]:
+    ) -> dict[str, Artifact]:
         """
         Produce the roc curve according to the ground truth (y) and predictions (y_pred) values. If predictions are not
         available, the model and a dataset can be given to produce them.

mlrun/frameworks/_ml_common/utils.py

@@ -14,7 +14,7 @@
 #
 from abc import ABC
 from enum import Enum
-from typing import Callable, List, Tuple, Union
+from typing import Callable, Union
 
 import pandas as pd
 from sklearn.base import is_classifier, is_regressor
@@ -137,10 +137,10 @@ class MLTypes(CommonTypes, ABC):
     # of the function and the full module path to the function to import. Arguments to use when calling the metric can
     # be joined by wrapping it as a tuple:
     # TODO: will be moved to SKLearn's framework once LightGBM and XGBoost are fully supported.
-    MetricEntryType = Union[Tuple[Union[Callable, str], dict], Callable, str]
+    MetricEntryType = Union[tuple[Union[Callable, str], dict], Callable, str]
 
     # Type for the target column name - a list of indices or column names that are the ground truth (y) of a dataset.
-    TargetColumnsNamesType = Union[List[str], List[int]]
+    TargetColumnsNamesType = Union[list[str], list[int]]
 
 
 class MLUtils(CommonUtils, ABC):
@@ -154,7 +154,7 @@ class MLUtils(CommonUtils, ABC):
         y: CommonTypes.DatasetType = None,
         target_columns_names: MLTypes.TargetColumnsNamesType = None,
         default_target_column_prefix: str = "y_",
-    ) -> Tuple[pd.DataFrame, Union[MLTypes.TargetColumnsNamesType, None]]:
+    ) -> tuple[pd.DataFrame, Union[MLTypes.TargetColumnsNamesType, None]]:
         """
         Concatenating the provided x and y data into a single pd.DataFrame, casting from np.ndarray and renaming y's
         original columns if 'y_columns' was not provided. The concatenated dataset index level will be reset to 0

mlrun/frameworks/auto_mlrun/auto_mlrun.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 #
 # flake8: noqa  - this is until we take care of the F401 violations with respect to __all__ & sphinx
-from typing import Callable, Dict, List, Tuple, Type, Union
+from typing import Callable, Union
 
 import mlrun
 from mlrun.artifacts import get_model
@@ -165,7 +165,7 @@ def get_framework_by_class_name(model: CommonTypes.ModelType) -> str:
     )
 
 
-def framework_to_model_handler(framework: str) -> Type[ModelHandler]:
+def framework_to_model_handler(framework: str) -> type[ModelHandler]:
     """
     Get the ModelHandler class of the given framework's name.
 
@@ -262,7 +262,7 @@ class AutoMLRun:
     @staticmethod
     def _get_framework(
         model: CommonTypes.ModelType = None, model_path: str = None
-    ) -> Union[Tuple[str, dict]]:
+    ) -> Union[tuple[str, dict]]:
         """
         Try to get the framework from the model or model path provided. The framework can be read from the model path
         only if the model path is of a logged model artifact (store object uri).
@@ -322,8 +322,8 @@ class AutoMLRun:
         model_path: str,
         model_name: str = None,
         context: mlrun.MLClientCtx = None,
-        modules_map: Union[Dict[str, Union[None, str, List[str]]], str] = None,
-        custom_objects_map: Union[Dict[str, Union[str, List[str]]], str] = None,
+        modules_map: Union[dict[str, Union[None, str, list[str]]], str] = None,
+        custom_objects_map: Union[dict[str, Union[str, list[str]]], str] = None,
         custom_objects_directory: str = None,
         framework: str = None,
         **kwargs,
@@ -363,7 +363,7 @@ class AutoMLRun:
 
                                              {
                                                  "/.../custom_model.py": "MyModel",
-                                                 "/.../custom_objects.py": ["object1", "object2"]
+                                                 "/.../custom_objects.py": ["object1", "object2"],
                                              }
 
                                          All the paths will be accessed from the given 'custom_objects_directory',
@@ -420,8 +420,8 @@ class AutoMLRun:
         model_name: str = None,
         tag: str = "",
         model_path: str = None,
-        modules_map: Union[Dict[str, Union[None, str, List[str]]], str] = None,
-        custom_objects_map: Union[Dict[str, Union[str, List[str]]], str] = None,
+        modules_map: Union[dict[str, Union[None, str, list[str]]], str] = None,
+        custom_objects_map: Union[dict[str, Union[str, list[str]]], str] = None,
         custom_objects_directory: str = None,
         context: mlrun.MLClientCtx = None,
         framework: str = None,
@@ -464,7 +464,7 @@ class AutoMLRun:
 
                                              {
                                                  "/.../custom_model.py": "MyModel",
-                                                 "/.../custom_objects.py": ["object1", "object2"]
+                                                 "/.../custom_objects.py": ["object1", "object2"],
                                              }
 
                                          All the paths will be accessed from the given 'custom_objects_directory',