mlrun 1.6.0rc15__py3-none-any.whl → 1.6.0rc16__py3-none-any.whl

mlrun/artifacts/base.py

@@ -712,7 +712,6 @@ class LinkArtifact(Artifact):
     category=FutureWarning,
 )
 class LegacyArtifact(ModelObj):
-
     _dict_fields = [
         "key",
         "kind",
@@ -930,7 +929,6 @@ class LegacyLinkArtifact(LegacyArtifact):
         link_key=None,
         link_tree=None,
     ):
-
         super().__init__(key)
         self.target_path = target_path
         self.link_iteration = link_iteration
@@ -958,7 +956,6 @@ def upload_extra_data(
         return
     target_path = artifact.target_path
     for key, item in extra_data.items():
-
         if isinstance(item, bytes):
             if target_path:
                 target = os.path.join(target_path, prefix + key)
@@ -1033,7 +1030,6 @@ def generate_target_path(item: Artifact, artifact_path, producer):
 
     suffix = "/"
     if not item.is_dir:
-
         # suffixes yields a list of suffixes, e.g. ['.tar', '.gz']
         # join them together to get the full suffix, e.g. '.tar.gz'
         suffix = "".join(pathlib.Path(item.src_path or "").suffixes)
@@ -1044,7 +1040,7 @@ def generate_target_path(item: Artifact, artifact_path, producer):
 
 
 def convert_legacy_artifact_to_new_format(
-    legacy_artifact: typing.Union[LegacyArtifact, dict]
+    legacy_artifact: typing.Union[LegacyArtifact, dict],
 ) -> Artifact:
     """Converts a legacy artifact to a new format.
 

mlrun/artifacts/dataset.py

@@ -59,7 +59,6 @@ class TableArtifact(Artifact):
         header=None,
         schema=None,
     ):
-
         if key:
             key_suffix = pathlib.Path(key).suffix
             if not format and key_suffix:
@@ -146,7 +145,6 @@ class DatasetArtifact(Artifact):
         label_column: str = None,
         **kwargs,
     ):
-
         format = (format or "").lower()
         super().__init__(key, None, format=format, target_path=target_path)
         if format and format not in self.SUPPORTED_FORMATS:
@@ -348,7 +346,6 @@ class LegacyTableArtifact(LegacyArtifact):
         header=None,
         schema=None,
     ):
-
         if key:
             key_suffix = pathlib.Path(key).suffix
             if not format and key_suffix:
@@ -417,7 +414,6 @@ class LegacyDatasetArtifact(LegacyArtifact):
         ignore_preview_limits: bool = False,
         **kwargs,
     ):
-
         format = (format or "").lower()
         super().__init__(key, None, format=format, target_path=target_path)
         if format and format not in self.SUPPORTED_FORMATS:

mlrun/artifacts/model.py

@@ -138,7 +138,6 @@ class ModelArtifact(Artifact):
         model_dir=None,
         **kwargs,
     ):
-
         super().__init__(key, body, format=format, target_path=target_path, **kwargs)
         model_file = str(model_file or "")
         if model_file and "/" in model_file:
@@ -309,7 +308,6 @@ class ModelArtifact(Artifact):
         spec_target_path = None
 
         if mlrun.mlconf.artifacts.generate_target_path_from_artifact_hash:
-
             # resolving target_path for the model spec
             _, spec_target_path = self.resolve_body_target_hash_path(
                 body=spec_body, artifact_path=artifact_path
@@ -337,7 +335,6 @@ class ModelArtifact(Artifact):
         artifact_path: str,
         target_model_path: str = None,
     ):
-
         body = self.spec.get_body()
         if body:
             if not target_model_path:
@@ -432,7 +429,6 @@ class LegacyModelArtifact(LegacyArtifact):
         model_target_file=None,
         **kwargs,
     ):
-
         super().__init__(key, body, format=format, target_path=target_path, **kwargs)
         self._inputs: ObjectList = None
         self._outputs: ObjectList = None
@@ -507,7 +503,6 @@ class LegacyModelArtifact(LegacyArtifact):
             self.labels["framework"] = self.framework
 
     def upload(self):
-
         target_model_path = path.join(self.target_path, self.model_file)
         body = self.get_body()
         if body:

mlrun/common/schemas/auth.py

@@ -71,10 +71,9 @@ class AuthorizationResourceTypes(mlrun.common.types.StrEnum):
             AuthorizationResourceTypes.project: "/projects/{project_name}",
             AuthorizationResourceTypes.function: "/projects/{project_name}/functions/{resource_name}",
             AuthorizationResourceTypes.artifact: "/projects/{project_name}/artifacts/{resource_name}",
-            # fmt: off
-            AuthorizationResourceTypes.project_background_task:
-                "/projects/{project_name}/background-tasks/{resource_name}",
-            # fmt: on
+            AuthorizationResourceTypes.project_background_task: (
+                "/projects/{project_name}/background-tasks/{resource_name}"
+            ),
             AuthorizationResourceTypes.background_task: "/background-tasks/{resource_name}",
             AuthorizationResourceTypes.feature_set: "/projects/{project_name}/feature-sets/{resource_name}",
             AuthorizationResourceTypes.feature_vector: "/projects/{project_name}/feature-vectors/{resource_name}",

mlrun/config.py

@@ -189,6 +189,7 @@ default_config = {
                 "migrations": "3600",
                 "load_project": "60",
                 "run_abortion": "600",
+                "abort_grace_period": "10",
             },
             "runtimes": {"dask": "600"},
         },

mlrun/data_types/to_pandas.py

@@ -61,7 +61,6 @@ def toPandas(spark_df):
             require_minimum_pyarrow_version()
             to_arrow_schema(spark_df.schema)
         except Exception as e:
-
             if spark_df.sql_ctx._conf.arrowPySparkFallbackEnabled():
                 msg = (
                     "toPandas attempted Arrow optimization because "

mlrun/datastore/base.py

@@ -317,7 +317,6 @@ class DataStore:
                     kwargs["storage_options"] = storage_options
                 df = reader(url, **kwargs)
             else:
-
                 file = url
                 # Workaround for ARROW-12472 affecting pyarrow 3.x and 4.x.
                 if file_system.protocol != "file":

mlrun/datastore/dbfs_store.py

@@ -123,7 +123,6 @@ class DBFSStore(DataStore):
         return self._filesystem.cat_file(key, start=start, end=end)
 
     def put(self, key, data, append=False):
-
         self._verify_filesystem_and_key(key)
         if append:
             raise mlrun.errors.MLRunInvalidArgumentError(

mlrun/datastore/sources.py

@@ -869,7 +869,7 @@ class StreamSource(OnlineSource):
         endpoint, stream_path = parse_path(self.path)
         v3io_client = v3io.dataplane.Client(endpoint=endpoint)
         container, stream_path = split_path(stream_path)
-        res = v3io_client.create_stream(
+        res = v3io_client.stream.create(
             container=container,
             path=stream_path,
             shard_count=self.attributes["shards"],

mlrun/datastore/v3io.py

@@ -175,7 +175,7 @@ class V3ioStore(DataStore):
         subpath_length = len(subpath) - 1
 
         try:
-            response = v3io_client.get_container_contents(
+            response = v3io_client.container.list(
                 container=container,
                 path=subpath,
                 get_all_attributes=False,

mlrun/datastore/wasbfs/fs.py

@@ -18,7 +18,6 @@ from fsspec import AbstractFileSystem
 
 
 class WasbFS(AbstractFileSystem):
-
     protocol = "wasb"
 
     def __init__(

mlrun/errors.py

@@ -44,7 +44,6 @@ class MLRunHTTPError(MLRunBaseError, requests.HTTPError):
         status_code: typing.Optional[int] = None,
         **kwargs,
     ):
-
         # because response object is probably with an error, it returns False, so we
         # should use 'is None' specifically
         if response is None:

mlrun/feature_store/retrieval/base.py

@@ -191,9 +191,7 @@ class BaseMerger(abc.ABC):
 
         feature_sets = []
         dfs = []
-        keys = (
-            []
-        )  # the struct of key is [[[],[]], ..] So that each record indicates which way the corresponding
+        keys = []  # the struct of key is [[[],[]], ..] So that each record indicates which way the corresponding
         # featureset is connected to the previous one, and within each record the left keys are indicated in index 0
         # and the right keys in index 1, this keys will be the keys that will be used in this join
         join_types = []
@@ -770,6 +768,7 @@ class BaseMerger(abc.ABC):
 
         raise mlrun.errors.MLRunRuntimeError("Failed to merge")
 
+    @classmethod
     def get_default_image(cls, kind):
         return mlrun.mlconf.feature_store.default_job_image
 

mlrun/feature_store/retrieval/job.py

@@ -79,7 +79,6 @@ def run_merge_job(
             )
         function.with_spark_service(spark_service=spark_service)
     elif run_config.kind == RuntimeKinds.spark:
-
         if mlconf.is_running_on_iguazio():
             function.with_igz_spark()
 

mlrun/feature_store/retrieval/spark_merger.py

@@ -49,7 +49,6 @@ class SparkFeatureMerger(BaseMerger):
         left_keys: list,
         right_keys: list,
     ):
-
         """Perform an as of join between entity and featureset.
         Join conditions:
         Args:
@@ -132,7 +131,6 @@ class SparkFeatureMerger(BaseMerger):
         left_keys: list,
         right_keys: list,
     ):
-
         """
         spark dataframes join
 

mlrun/feature_store/steps.py

@@ -411,7 +411,6 @@ class Imputer(StepToDict, MLRunStep):
         return event
 
     def _do_spark(self, event):
-
         for feature in event.columns:
             val = self.mapping.get(feature, self.default_value)
             if val is not None:
@@ -451,7 +450,6 @@ class OneHotEncoder(StepToDict, MLRunStep):
         encoding = self.mapping.get(feature, [])
 
         if encoding:
-
             one_hot_encoding = {
                 f"{feature}_{OneHotEncoder._sanitized_category(category)}": 0
                 for category in encoding
@@ -476,7 +474,6 @@ class OneHotEncoder(StepToDict, MLRunStep):
         return encoded_values
 
     def _do_pandas(self, event):
-
         for key, values in self.mapping.items():
             event[key] = pd.Categorical(event[key], categories=list(values))
             encoded = pd.get_dummies(event[key], prefix=key, dtype=np.int64)

mlrun/frameworks/_common/model_handler.py

@@ -157,9 +157,7 @@ class ModelHandler(ABC, Generic[CommonTypes.ModelType, CommonTypes.IOSampleType]
         self._model_artifact = kwargs.get("model_artifact", None)  # type: ModelArtifact
 
         # If the model path is of a store model object, this will be the extra data as DataItems ready to be downloaded.
-        self._extra_data = kwargs.get(
-            "extra_data", {}
-        )  # type: Dict[str, CommonTypes.ExtraDataType]
+        self._extra_data = kwargs.get("extra_data", {})  # type: Dict[str, CommonTypes.ExtraDataType]
 
         # If the model key is passed, override the default:
         self._model_key = kwargs.get("model_key", "model")
@@ -1064,7 +1062,7 @@ class ModelHandler(ABC, Generic[CommonTypes.ModelType, CommonTypes.IOSampleType]
 
     @staticmethod
     def _validate_modules_parameter(
-        modules_map: Union[Dict[str, Union[None, str, List[str]]], str]
+        modules_map: Union[Dict[str, Union[None, str, List[str]]], str],
     ):
         """
         Validate the given modules parameter.

mlrun/frameworks/_dl_common/loggers/logger.py

@@ -53,9 +53,7 @@ class Logger:
 
         # Setup the dynamic hyperparameters dictionary - a dictionary of all tracked hyperparameters by epochs:
         # [Hyperparameter: str] -> [Epoch: int] -> [value: Union[str, bool, float, int]]
-        self._dynamic_hyperparameters = (
-            {}
-        )  # type: Dict[str, List[DLTypes.TrackableType]]
+        self._dynamic_hyperparameters = {}  # type: Dict[str, List[DLTypes.TrackableType]]
 
         # Setup the iterations counter:
         self._epochs = 0

mlrun/frameworks/_dl_common/loggers/tensorboard_logger.py

@@ -133,9 +133,7 @@ class TensorboardLogger(Logger, Generic[DLTypes.WeightType]):
         # [Statistic: str] -> [Weight: str] -> [epoch: int] -> [value: float]
         self._weights_statistics = {}  # type: Dict[str, Dict[str, List[float]]]
         for statistic_function in self._statistics_functions:
-            self._weights_statistics[
-                statistic_function.__name__
-            ] = {}  # type: Dict[str, List[float]]
+            self._weights_statistics[statistic_function.__name__] = {}  # type: Dict[str, List[float]]
 
     @property
     def weights(self) -> Dict[str, DLTypes.WeightType]:

mlrun/frameworks/_ml_common/loggers/logger.py

@@ -36,9 +36,7 @@ class Logger:
 
         # Set up the dynamic hyperparameters dictionary - a dictionary of all tracked hyperparameters by epochs:
         # [Hyperparameter: str] -> [Epoch: int] -> [value: Union[str, bool, float, int]]
-        self._dynamic_hyperparameters = (
-            {}
-        )  # type: Dict[str, List[MLTypes.TrackableType]]
+        self._dynamic_hyperparameters = {}  # type: Dict[str, List[MLTypes.TrackableType]]
 
         # Set up the iterations counter:
         self._iterations = 0

mlrun/frameworks/_ml_common/plans/calibration_curve_plan.py

@@ -72,7 +72,7 @@ class CalibrationCurvePlan(MLPlotPlan):
         y_pred: MLTypes.DatasetType = None,
         model: MLTypes.ModelType = None,
         x: MLTypes.DatasetType = None,
-        **kwargs
+        **kwargs,
     ) -> Dict[str, Artifact]:
         """
         Produce the calibration curve according to the ground truth (y) and predictions (y_pred) values. If predictions

mlrun/frameworks/_ml_common/plans/confusion_matrix_plan.py

@@ -78,7 +78,7 @@ class ConfusionMatrixPlan(MLPlotPlan):
         y_pred: MLTypes.DatasetType = None,
         model: MLTypes.ModelType = None,
         x: MLTypes.DatasetType = None,
-        **kwargs
+        **kwargs,
     ) -> Dict[str, Artifact]:
         """
         Produce the confusion matrix according to the ground truth (y) and predictions (y_pred) values. If predictions

mlrun/frameworks/_ml_common/plans/dataset_plan.py

@@ -89,9 +89,7 @@ class DatasetPlan(MLPlan):
         self._preview = preview
         self._stats = stats
         self._fmt = fmt
-        self._plans = (
-            {}
-        )  # TODO: Implement DatasetPlansLibrary with dataset specific artifacts plans.
+        self._plans = {}  # TODO: Implement DatasetPlansLibrary with dataset specific artifacts plans.
 
         # Continue initializing the plan:
         super(DatasetPlan, self).__init__(need_probabilities=False)

mlrun/frameworks/lgbm/__init__.py

@@ -106,7 +106,7 @@ def _apply_mlrun_on_model(
     parameters: Dict[str, Union[str, int, float]] = None,
     extra_data: Dict[str, LGBMTypes.ExtraDataType] = None,
     auto_log: bool = True,
-    **kwargs
+    **kwargs,
 ):
     # Create a model handler:
     model_handler_kwargs = (
@@ -205,7 +205,7 @@ def apply_mlrun(
     extra_data: Dict[str, LGBMTypes.ExtraDataType] = None,
     auto_log: bool = True,
     mlrun_logging_callback_kwargs: Dict[str, Any] = None,
-    **kwargs
+    **kwargs,
 ) -> Union[LGBMModelHandler, None]:
     """
     Apply MLRun's interface on top of LightGBM by wrapping the module itself or the given model, providing both with

mlrun/frameworks/pytorch/callbacks/tensorboard_logging_callback.py

@@ -364,9 +364,9 @@ class TensorboardLoggingCallback(LoggingCallback):
         return self._logger.weight_statistics
 
     @staticmethod
-    def get_default_weight_statistics_list() -> List[
-        Callable[[Union[Parameter, Tensor]], Union[float, Tensor]]
-    ]:
+    def get_default_weight_statistics_list() -> (
+        List[Callable[[Union[Parameter, Tensor]], Union[float, Tensor]]]
+    ):
         """
         Get the default list of statistics functions being applied on the tracked weights each epoch.
 

mlrun/frameworks/pytorch/mlrun_interface.py

@@ -949,7 +949,7 @@ class PyTorchMLRunInterface:
 
     @staticmethod
     def _tensor_to_cuda(
-        tensor: Union[Tensor, Dict, List, Tuple]
+        tensor: Union[Tensor, Dict, List, Tuple],
     ) -> Union[Tensor, Dict, List, Tuple]:
         """
         Send to given tensor to cuda if it is a tensor. If the given object is a dictionary, the dictionary values will

mlrun/frameworks/tf_keras/__init__.py

@@ -42,7 +42,7 @@ def apply_mlrun(
     mlrun_callback_kwargs: Dict[str, Any] = None,
     tensorboard_callback_kwargs: Dict[str, Any] = None,
     use_horovod: bool = None,
-    **kwargs
+    **kwargs,
 ) -> TFKerasModelHandler:
     """
     Wrap the given model with MLRun's interface providing it with mlrun's additional features.
@@ -143,7 +143,7 @@ def apply_mlrun(
         modules_map=modules_map,
         custom_objects_map=custom_objects_map,
         custom_objects_directory=custom_objects_directory,
-        **model_handler_kwargs
+        **model_handler_kwargs,
     )
 
     # Load the model if it was not provided:
@@ -174,7 +174,7 @@ def apply_mlrun(
                 model_handler=handler,
                 log_model_tag=tag,
                 auto_log=auto_log,
-                **mlrun_callback_kwargs
+                **mlrun_callback_kwargs,
             )
         )
         model.add_logging_callback(
@@ -182,7 +182,7 @@ def apply_mlrun(
                 context=context,
                 tensorboard_directory=tensorboard_directory,
                 auto_log=auto_log,
-                **tensorboard_callback_kwargs
+                **tensorboard_callback_kwargs,
             )
         )
 

mlrun/frameworks/tf_keras/callbacks/tensorboard_logging_callback.py

@@ -554,9 +554,9 @@ class TensorboardLoggingCallback(LoggingCallback):
             self._logger.write_dynamic_hyperparameters()
 
     @staticmethod
-    def get_default_weight_statistics_list() -> List[
-        Callable[[Union[Variable, Tensor]], Union[float, Tensor]]
-    ]:
+    def get_default_weight_statistics_list() -> (
+        List[Callable[[Union[Variable, Tensor]], Union[float, Tensor]]]
+    ):
         """
         Get the default list of statistics functions being applied on the tracked weights each epoch.
 

mlrun/frameworks/tf_keras/mlrun_interface.py

@@ -148,7 +148,12 @@ class TFKerasMLRunInterface(MLRunInterface, ABC):
             kwargs["validation_data"] = kwargs.get("validation_data", None)
 
             # Call the pre fit method:
-            (callbacks, verbose, steps_per_epoch, validation_steps,) = self._pre_fit(
+            (
+                callbacks,
+                verbose,
+                steps_per_epoch,
+                validation_steps,
+            ) = self._pre_fit(
                 callbacks=kwargs["callbacks"],
                 verbose=kwargs["verbose"],
                 steps_per_epoch=kwargs["steps_per_epoch"],

mlrun/frameworks/xgboost/__init__.py

@@ -55,7 +55,7 @@ def apply_mlrun(
     parameters: Dict[str, Union[str, int, float]] = None,
     extra_data: Dict[str, XGBoostTypes.ExtraDataType] = None,
     auto_log: bool = True,
-    **kwargs
+    **kwargs,
 ) -> XGBoostModelHandler:
     """
     Wrap the given model with MLRun's interface providing it with mlrun's additional features.

mlrun/k8s_utils.py

@@ -60,9 +60,9 @@ def generate_preemptible_node_selector_requirements(
     return match_expressions
 
 
-def generate_preemptible_nodes_anti_affinity_terms() -> typing.List[
-    kubernetes.client.V1NodeSelectorTerm
-]:
+def generate_preemptible_nodes_anti_affinity_terms() -> (
+    typing.List[kubernetes.client.V1NodeSelectorTerm]
+):
     """
     Generate node selector term containing anti-affinity expressions based on the
     pre-configured node selector of the preemptible nodes.
@@ -82,9 +82,9 @@ def generate_preemptible_nodes_anti_affinity_terms() -> typing.List[
     ]
 
 
-def generate_preemptible_nodes_affinity_terms() -> typing.List[
-    kubernetes.client.V1NodeSelectorTerm
-]:
+def generate_preemptible_nodes_affinity_terms() -> (
+    typing.List[kubernetes.client.V1NodeSelectorTerm]
+):
     """
     Use for purpose of scheduling on node having at least one of the node selectors.
     When specifying multiple nodeSelectorTerms associated with nodeAffinity types,

mlrun/kfpops.py

@@ -313,7 +313,6 @@ def mlrun_op(
     code_env = None
     function_name = ""
     if function:
-
         if not func_url:
             if function.kind in ["", "local"]:
                 image = image or function.spec.image
@@ -492,7 +491,6 @@ def deploy_op(
     tag="",
     verbose=False,
 ):
-
     cmd = ["python", "-m", "mlrun", "deploy"]
     if source:
         cmd += ["-s", source]
@@ -855,7 +853,6 @@ def add_default_function_resources(
 def add_function_node_selection_attributes(
     function, container_op: dsl.ContainerOp
 ) -> dsl.ContainerOp:
-
     if not mlrun.runtimes.RuntimeKinds.is_local_runtime(function.kind):
         if getattr(function.spec, "node_selector"):
             container_op.node_selector = function.spec.node_selector

mlrun/launcher/base.py

@@ -177,7 +177,6 @@ class BaseLauncher(abc.ABC):
 
     def _validate_run_params(self, parameters: Dict[str, Any]):
         for param_name, param_value in parameters.items():
-
             if isinstance(param_value, dict):
                 # if the parameter is a dict, we might have some nested parameters,
                 # in this case we need to verify them as well recursively

mlrun/launcher/local.py

@@ -69,7 +69,6 @@ class ClientLocalLauncher(launcher.ClientBaseLauncher):
         returns: Optional[List[Union[str, Dict[str, str]]]] = None,
         state_thresholds: Optional[Dict[str, int]] = None,
     ) -> "mlrun.run.RunObject":
-
         # do not allow local function to be scheduled
         if self._is_run_local and schedule is not None:
             raise mlrun.errors.MLRunInvalidArgumentError(
@@ -133,7 +132,6 @@ class ClientLocalLauncher(launcher.ClientBaseLauncher):
         runtime: "mlrun.runtimes.BaseRuntime",
         run: Optional[Union["mlrun.run.RunTemplate", "mlrun.run.RunObject"]] = None,
     ):
-
         if "V3IO_USERNAME" in os.environ and "v3io_user" not in run.metadata.labels:
             run.metadata.labels["v3io_user"] = os.environ.get("V3IO_USERNAME")
 
@@ -209,7 +207,6 @@ class ClientLocalLauncher(launcher.ClientBaseLauncher):
         workdir: Optional[str] = "",
         handler: Optional[str] = None,
     ):
-
         project = project or runtime.metadata.project
         function_name = name or runtime.metadata.name
         command, args = self._resolve_local_code_path(local_code_path)

mlrun/model.py

@@ -1258,13 +1258,14 @@ class RunObject(RunTemplate):
     def error(self) -> str:
         """error string if failed"""
         if self.status:
-            if self.status.state != "error":
-                return f"Run state ({self.status.state}) is not in error state"
+            unknown_error = ""
+            if self.status.state in mlrun.runtimes.constants.RunStates.error_states():
+                unknown_error = "Unknown error"
             return (
                 self.status.error
                 or self.status.reason
                 or self.status.status_text
-                or "Unknown error"
+                or unknown_error
             )
         return ""
 

mlrun/model_monitoring/batch.py

@@ -526,12 +526,14 @@ class BatchProcessor:
         )
 
         # Get drift thresholds from the model monitoring configuration
+        # fmt: off
         self.default_possible_drift_threshold = (
             mlrun.mlconf.model_endpoint_monitoring.drift_thresholds.default.possible_drift
         )
         self.default_drift_detected_threshold = (
             mlrun.mlconf.model_endpoint_monitoring.drift_thresholds.default.drift_detected
         )
+        # fmt: on
 
         # Get a runtime database
 
@@ -618,7 +620,7 @@ class BatchProcessor:
 
         if not mlrun.mlconf.is_ce_mode():
             # Create v3io stream based on the input stream
-            response = self.v3io.create_stream(
+            response = self.v3io.stream.create(
                 container=self.stream_container,
                 path=self.stream_path,
                 shard_count=1,

mlrun/package/packagers/numpy_packagers.py

@@ -514,7 +514,7 @@ class _NumPyNDArrayCollectionPackager(DefaultPackager):
 
     @staticmethod
     def _is_any_object_dtype(
-        array_collection: Union[np.ndarray, NumPyArrayCollectionType]
+        array_collection: Union[np.ndarray, NumPyArrayCollectionType],
     ):
         """
         Check if any of the arrays in a collection is of type `object`.

mlrun/package/utils/log_hint_utils.py

@@ -35,7 +35,7 @@ class LogHintUtils:
 
     @staticmethod
     def parse_log_hint(
-        log_hint: typing.Union[typing.Dict[str, str], str, None]
+        log_hint: typing.Union[typing.Dict[str, str], str, None],
     ) -> typing.Union[typing.Dict[str, str], None]:
         """
         Parse a given log hint from string to a logging configuration dictionary. The string will be read as the

mlrun/package/utils/type_hint_utils.py

@@ -249,7 +249,9 @@ class TypeHintUtils:
                 if type_hint.__forward_module__:
                     arg = f"{type_hint.__forward_module__}.{arg}"
                 return [TypeHintUtils.parse_type_hint(type_hint=arg)]
-            except MLRunInvalidArgumentError:  # May be raised from `TypeHintUtils.parse_type_hint`
+            except (
+                MLRunInvalidArgumentError
+            ):  # May be raised from `TypeHintUtils.parse_type_hint`
                 logger.warn(
                     f"Could not reduce the type hint '{type_hint}' as it is a forward reference to a class without "
                     f"it's full module path. To enable importing forward references, please provide the full module "