mlrun 1.7.1rc10__py3-none-any.whl → 1.8.0rc8__py3-none-any.whl

mlrun/feature_store/feature_set.py

@@ -323,14 +323,14 @@ class FeatureSet(ModelObj):
 
     def __init__(
         self,
-        name: str = None,
-        description: str = None,
-        entities: list[Union[Entity, str]] = None,
-        timestamp_key: str = None,
-        engine: str = None,
-        label_column: str = None,
-        relations: dict[str, Union[Entity, str]] = None,
-        passthrough: bool = None,
+        name: Optional[str] = None,
+        description: Optional[str] = None,
+        entities: Optional[list[Union[Entity, str]]] = None,
+        timestamp_key: Optional[str] = None,
+        engine: Optional[str] = None,
+        label_column: Optional[str] = None,
+        relations: Optional[dict[str, Union[Entity, str]]] = None,
+        passthrough: Optional[bool] = None,
     ):
         """Feature set object, defines a set of features and their data pipeline
 
@@ -533,7 +533,9 @@ class FeatureSet(ModelObj):
                     self, **(class_args if class_args is not None else {})
                 )
 
-    def purge_targets(self, target_names: list[str] = None, silent: bool = False):
+    def purge_targets(
+        self, target_names: Optional[list[str]] = None, silent: bool = False
+    ):
         """Delete data of specific targets
         :param target_names: List of names of targets to delete (default: delete all ingested targets)
         :param silent: Fail silently if target doesn't exist in featureset status"""
@@ -562,7 +564,7 @@ class FeatureSet(ModelObj):
     def update_targets_for_ingest(
         self,
         targets: list[DataTargetBase],
-        overwrite: bool = None,
+        overwrite: Optional[bool] = None,
     ):
         if not targets:
             return
@@ -582,7 +584,7 @@ class FeatureSet(ModelObj):
         update_targets_run_id_for_ingest(overwrite, targets, status_targets)
 
     def _reload_and_get_status_targets(
-        self, target_names: list[str] = None, silent: bool = False
+        self, target_names: Optional[list[str]] = None, silent: bool = False
     ):
         try:
             self.reload(update_spec=False)
@@ -619,7 +621,7 @@ class FeatureSet(ModelObj):
         self,
         name: str,
         value_type: mlrun.data_types.ValueType = None,
-        description: str = None,
+        description: Optional[str] = None,
         labels: Optional[dict[str, str]] = None,
     ):
         """add/set an entity (dataset index)
@@ -1004,7 +1006,7 @@ class FeatureSet(ModelObj):
     def ingest(
         self,
         source=None,
-        targets: list[DataTargetBase] = None,
+        targets: Optional[list[DataTargetBase]] = None,
         namespace=None,
         return_df: bool = True,
         infer_options: InferOptions = InferOptions.default(),
@@ -1073,11 +1075,11 @@ class FeatureSet(ModelObj):
     def preview(
         self,
         source,
-        entity_columns: list = None,
+        entity_columns: Optional[list] = None,
         namespace=None,
         options: InferOptions = None,
         verbose: bool = False,
-        sample_size: int = None,
+        sample_size: Optional[int] = None,
     ) -> pd.DataFrame:
         """run the ingestion pipeline with local DataFrame/file data and infer features schema and stats
 
@@ -1106,8 +1108,8 @@ class FeatureSet(ModelObj):
     def deploy_ingestion_service(
         self,
         source: DataSource = None,
-        targets: list[DataTargetBase] = None,
-        name: str = None,
+        targets: Optional[list[DataTargetBase]] = None,
+        name: Optional[str] = None,
         run_config: RunConfig = None,
         verbose=False,
     ) -> tuple[str, BaseRuntime]:
@@ -1143,7 +1145,7 @@ class FeatureSet(ModelObj):
     def extract_relation_keys(
         self,
         other_feature_set,
-        relations: dict[str, Union[str, Entity]] = None,
+        relations: Optional[dict[str, Union[str, Entity]]] = None,
     ) -> list[str]:
         """
         Checks whether a feature set can be merged to the right of this feature set.

mlrun/feature_store/feature_vector.py

@@ -212,7 +212,7 @@ class JoinGraph(ModelObj):
 
     def __init__(
         self,
-        name: str = None,
+        name: typing.Optional[str] = None,
         first_feature_set: Union[str, FeatureSet] = None,
     ):
         """
@@ -318,7 +318,10 @@ class JoinGraph(ModelObj):
         return self._join_operands(other_operand, JoinGraph.first_join_type)
 
     def _init_all_join_keys(
-        self, feature_set_objects, vector, entity_rows_keys: list[str] = None
+        self,
+        feature_set_objects,
+        vector,
+        entity_rows_keys: typing.Optional[list[str]] = None,
     ):
         for step in self.steps:
             step.init_join_keys(feature_set_objects, vector, entity_rows_keys)
@@ -371,11 +374,11 @@ class JoinGraph(ModelObj):
 class _JoinStep(ModelObj):
     def __init__(
         self,
-        name: str = None,
-        left_step_name: str = None,
-        right_step_name: str = None,
-        left_feature_set_names: Union[str, list[str]] = None,
-        right_feature_set_name: str = None,
+        name: typing.Optional[str] = None,
+        left_step_name: typing.Optional[str] = None,
+        right_step_name: typing.Optional[str] = None,
+        left_feature_set_names: typing.Optional[Union[str, list[str]]] = None,
+        right_feature_set_name: typing.Optional[str] = None,
         join_type: str = "inner",
         asof_join: bool = False,
     ):
@@ -399,7 +402,7 @@ class _JoinStep(ModelObj):
         self,
         feature_set_objects: ObjectList,
         vector,
-        entity_rows_keys: list[str] = None,
+        entity_rows_keys: typing.Optional[list[str]] = None,
     ):
         if feature_set_objects[self.right_feature_set_name].is_connectable_to_df(
             entity_rows_keys
@@ -479,7 +482,7 @@ class FeatureVector(ModelObj):
         description=None,
         with_indexes=None,
         join_graph: JoinGraph = None,
-        relations: dict[str, dict[str, Union[Entity, str]]] = None,
+        relations: typing.Optional[dict[str, dict[str, Union[Entity, str]]]] = None,
     ):
         """Feature vector, specify selected features, their metadata and material views
 
@@ -727,21 +730,21 @@ class FeatureVector(ModelObj):
     def get_offline_features(
         self,
         entity_rows=None,
-        entity_timestamp_column: str = None,
+        entity_timestamp_column: typing.Optional[str] = None,
         target: DataTargetBase = None,
         run_config: RunConfig = None,
-        drop_columns: list[str] = None,
-        start_time: Union[str, datetime] = None,
-        end_time: Union[str, datetime] = None,
+        drop_columns: typing.Optional[list[str]] = None,
+        start_time: typing.Optional[Union[str, datetime]] = None,
+        end_time: typing.Optional[Union[str, datetime]] = None,
         with_indexes: bool = False,
         update_stats: bool = False,
-        engine: str = None,
-        engine_args: dict = None,
-        query: str = None,
-        order_by: Union[str, list[str]] = None,
-        spark_service: str = None,
-        timestamp_for_filtering: Union[str, dict[str, str]] = None,
-        additional_filters: list = None,
+        engine: typing.Optional[str] = None,
+        engine_args: typing.Optional[dict] = None,
+        query: typing.Optional[str] = None,
+        order_by: typing.Optional[Union[str, list[str]]] = None,
+        spark_service: typing.Optional[str] = None,
+        timestamp_for_filtering: typing.Optional[Union[str, dict[str, str]]] = None,
+        additional_filters: typing.Optional[list] = None,
     ):
         """retrieve offline feature vector results
 
@@ -831,9 +834,9 @@ class FeatureVector(ModelObj):
         self,
         run_config: RunConfig = None,
         fixed_window_type: FixedWindowType = FixedWindowType.LastClosedWindow,
-        impute_policy: dict = None,
+        impute_policy: typing.Optional[dict] = None,
         update_stats: bool = False,
-        entity_keys: list[str] = None,
+        entity_keys: typing.Optional[list[str]] = None,
     ):
         """initialize and return online feature vector service api,
         returns :py:class:`~mlrun.feature_store.OnlineVectorService`
@@ -915,8 +918,8 @@ class OnlineVectorService:
         vector,
         graph,
         index_columns,
-        impute_policy: dict = None,
-        requested_columns: list[str] = None,
+        impute_policy: typing.Optional[dict] = None,
+        requested_columns: typing.Optional[list[str]] = None,
     ):
         self.vector = vector
         self.impute_policy = impute_policy or {}

mlrun/feature_store/retrieval/base.py

@@ -64,7 +64,12 @@ class BaseMerger(abc.ABC):
             if self._drop_indexes:
                 self._append_drop_column(key)
 
-    def _update_alias(self, key: str = None, val: str = None, dictionary: dict = None):
+    def _update_alias(
+        self,
+        key: typing.Optional[str] = None,
+        val: typing.Optional[str] = None,
+        dictionary: typing.Optional[dict] = None,
+    ):
         if dictionary is not None:
             # adding dictionary to alias
             self._alias.update(dictionary)
@@ -414,8 +419,8 @@ class BaseMerger(abc.ABC):
         entity_timestamp_column: str,
         featuresets: list,
         featureset_dfs: list,
-        keys: list = None,
-        join_types: list = None,
+        keys: typing.Optional[list] = None,
+        join_types: typing.Optional[list] = None,
     ):
         """join the entities and feature set features into a result dataframe"""
 
@@ -544,8 +549,8 @@ class BaseMerger(abc.ABC):
             self,
             name: str,
             order: int,
-            left_keys: list[str] = None,
-            right_keys: list[str] = None,
+            left_keys: typing.Optional[list[str]] = None,
+            right_keys: typing.Optional[list[str]] = None,
         ):
             self.name = name
             self.left_keys = left_keys if left_keys is not None else []
@@ -755,9 +760,9 @@ class BaseMerger(abc.ABC):
         self,
         feature_set: FeatureSet,
         feature_set_name: list[str],
-        column_names: list[str] = None,
-        start_time: typing.Union[str, datetime] = None,
-        end_time: typing.Union[str, datetime] = None,
+        column_names: typing.Optional[list[str]] = None,
+        start_time: typing.Optional[typing.Union[str, datetime]] = None,
+        end_time: typing.Optional[typing.Union[str, datetime]] = None,
         time_column: typing.Optional[str] = None,
         additional_filters=None,
     ):
@@ -779,7 +784,7 @@ class BaseMerger(abc.ABC):
         self,
         df,
         rename_col_dict: dict[str, str],
-        columns: list[str] = None,
+        columns: typing.Optional[list[str]] = None,
     ):
         """
         rename the columns of the df according to rename_col_dict, and select only `columns` if it is not none

mlrun/feature_store/retrieval/job.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 #
 import uuid
+from typing import Optional
 
 import mlrun
 import mlrun.common.constants as mlrun_constants
@@ -32,7 +33,7 @@ def run_merge_job(
     merger: BaseMerger,
     engine: str,
     engine_args: dict,
-    spark_service: str = None,
+    spark_service: Optional[str] = None,
     entity_rows=None,
     entity_timestamp_column=None,
     run_config=None,

mlrun/feature_store/steps.py

@@ -379,7 +379,7 @@ class Imputer(StepToDict, MLRunStep):
         self,
         method: str = "avg",
         default_value=None,
-        mapping: dict[str, Any] = None,
+        mapping: Optional[dict[str, Any]] = None,
         **kwargs,
     ):
         """Replace None values with default values
@@ -517,7 +517,7 @@ class DateExtractor(StepToDict, MLRunStep):
     def __init__(
         self,
         parts: Union[dict[str, str], list[str]],
-        timestamp_col: str = None,
+        timestamp_col: Optional[str] = None,
         **kwargs,
     ):
         """Date Extractor extracts a date-time component into new columns

mlrun/features.py

@@ -41,9 +41,9 @@ class Entity(ModelObj):
 
     def __init__(
         self,
-        name: str = None,
+        name: Optional[str] = None,
         value_type: Union[ValueType, str] = None,
-        description: str = None,
+        description: Optional[str] = None,
         labels: Optional[dict[str, str]] = None,
     ):
         """data entity (index key)
@@ -80,13 +80,13 @@ class Feature(ModelObj):
     def __init__(
         self,
         value_type: Union[ValueType, str] = None,
-        dims: list[int] = None,
-        description: str = None,
-        aggregate: bool = None,
-        name: str = None,
+        dims: Optional[list[int]] = None,
+        description: Optional[str] = None,
+        aggregate: Optional[bool] = None,
+        name: Optional[str] = None,
         validator=None,
-        default: str = None,
-        labels: dict[str, str] = None,
+        default: Optional[str] = None,
+        labels: Optional[dict[str, str]] = None,
     ):
         """data feature
 
@@ -231,7 +231,9 @@ class Validator(ModelObj):
     kind = ""
     _dict_fields = ["kind", "check_type", "severity"]
 
-    def __init__(self, check_type: bool = None, severity: str = None):
+    def __init__(
+        self, check_type: Optional[bool] = None, severity: Optional[str] = None
+    ):
         """Base validator
 
         example::
@@ -268,7 +270,11 @@ class MinMaxValidator(Validator):
     _dict_fields = Validator._dict_fields + ["min", "max"]
 
     def __init__(
-        self, check_type: bool = None, severity: str = None, min=None, max=None
+        self,
+        check_type: Optional[bool] = None,
+        severity: Optional[str] = None,
+        min=None,
+        max=None,
     ):
         """Validate min/max value ranges
 
@@ -328,7 +334,11 @@ class MinMaxLenValidator(Validator):
     _dict_fields = Validator._dict_fields + ["min", "max"]
 
     def __init__(
-        self, check_type: bool = None, severity: str = None, min=None, max=None
+        self,
+        check_type: Optional[bool] = None,
+        severity: Optional[str] = None,
+        min=None,
+        max=None,
     ):
         """Validate min/max length value ranges
 
@@ -390,7 +400,12 @@ class RegexValidator(Validator):
     kind = "regex"
     _dict_fields = Validator._dict_fields + ["regex"]
 
-    def __init__(self, check_type: bool = None, severity: str = None, regex=None):
+    def __init__(
+        self,
+        check_type: Optional[bool] = None,
+        severity: Optional[str] = None,
+        regex=None,
+    ):
         """Validate value based on regular expression
 
         example::
@@ -434,7 +449,9 @@ class RegexValidator(Validator):
         return ok, args
 
     @classmethod
-    def from_dict(cls, struct=None, fields=None, deprecated_fields: dict = None):
+    def from_dict(
+        cls, struct=None, fields=None, deprecated_fields: Optional[dict] = None
+    ):
         new_obj = super().from_dict(
             struct=struct, fields=fields, deprecated_fields=deprecated_fields
         )

mlrun/frameworks/__init__.py

@@ -11,6 +11,5 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-#
-# flake8: noqa  - this is until we take care of the F401 violations with respect to __all__ & sphinx
+
 from .parallel_coordinates import compare_db_runs, compare_run_objects

mlrun/frameworks/_common/__init__.py

@@ -11,8 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-#
-# flake8: noqa  - this is until we take care of the F401 violations with respect to __all__ & sphinx
+
 from .artifacts_library import ArtifactsLibrary
 from .mlrun_interface import MLRunInterface
 from .model_handler import ModelHandler, with_mlrun_interface, without_mlrun_interface

mlrun/frameworks/_common/artifacts_library.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 #
 from abc import ABC, abstractmethod
-from typing import Union
+from typing import Optional, Union
 
 import mlrun
 
@@ -39,7 +39,7 @@ class ArtifactsLibrary(ABC):
     @classmethod
     def get_plans(
         cls,
-        artifacts: Union[list[Plan], dict[str, dict], list[str]] = None,
+        artifacts: Optional[Union[list[Plan], dict[str, dict], list[str]]] = None,
         context: mlrun.MLClientCtx = None,
         include_default: bool = True,
         # custom_plans: dict = None, :param custom_plans: Custom user plans objects to initialize from.

mlrun/frameworks/_common/mlrun_interface.py

@@ -17,7 +17,7 @@ import functools
 import inspect
 from abc import ABC
 from types import FunctionType, MethodType
-from typing import Any, Generic, Union
+from typing import Any, Generic, Optional, Union
 
 from .utils import CommonTypes
 
@@ -173,7 +173,7 @@ class MLRunInterface(ABC, Generic[CommonTypes.MLRunInterfaceableType]):
     def _insert_properties(
         cls,
         obj: CommonTypes.MLRunInterfaceableType,
-        properties: dict[str, Any] = None,
+        properties: Optional[dict[str, Any]] = None,
     ):
         """
         Insert the properties of the interface to the object. The properties default values are being copied (not deep
@@ -238,7 +238,9 @@ class MLRunInterface(ABC, Generic[CommonTypes.MLRunInterfaceableType]):
 
     @classmethod
     def _replace_properties(
-        cls, obj: CommonTypes.MLRunInterfaceableType, properties: dict[str, Any] = None
+        cls,
+        obj: CommonTypes.MLRunInterfaceableType,
+        properties: Optional[dict[str, Any]] = None,
     ):
         """
         Replace the properties of the given object according to the configuration in the MLRun interface.
@@ -282,7 +284,9 @@ class MLRunInterface(ABC, Generic[CommonTypes.MLRunInterfaceableType]):
 
     @classmethod
     def _replace_functions(
-        cls, obj: CommonTypes.MLRunInterfaceableType, functions: list[str] = None
+        cls,
+        obj: CommonTypes.MLRunInterfaceableType,
+        functions: Optional[list[str]] = None,
     ):
         """
         Replace the functions / methods of the given object according to the configuration in the MLRun interface.
@@ -417,8 +421,8 @@ class MLRunInterface(ABC, Generic[CommonTypes.MLRunInterfaceableType]):
     def _get_function_argument(
         func: FunctionType,
         argument_name: str,
-        passed_args: tuple = None,
-        passed_kwargs: dict = None,
+        passed_args: Optional[tuple] = None,
+        passed_kwargs: Optional[dict] = None,
         default_value: Any = None,
     ) -> tuple[Any, Union[str, int, None]]:
         """

mlrun/frameworks/_common/model_handler.py

@@ -20,7 +20,7 @@ import sys
 import zipfile
 from abc import ABC, abstractmethod
 from types import MethodType
-from typing import Any, Generic, Union
+from typing import Any, Generic, Optional, Union
 
 import numpy as np
 
@@ -55,7 +55,7 @@ class ModelHandler(ABC, Generic[CommonTypes.ModelType, CommonTypes.IOSampleType]
         self,
         model: CommonTypes.ModelType = None,
         model_path: CommonTypes.PathType = None,
-        model_name: str = None,
+        model_name: Optional[str] = None,
         modules_map: Union[
             dict[str, Union[None, str, list[str]]], CommonTypes.PathType
         ] = None,
@@ -306,7 +306,7 @@ class ModelHandler(ABC, Generic[CommonTypes.ModelType, CommonTypes.IOSampleType]
     def set_inputs(
         self,
         from_sample: CommonTypes.IOSampleType = None,
-        features: list[Feature] = None,
+        features: Optional[list[Feature]] = None,
         **kwargs,
     ):
         """
@@ -335,7 +335,7 @@ class ModelHandler(ABC, Generic[CommonTypes.ModelType, CommonTypes.IOSampleType]
     def set_outputs(
         self,
         from_sample: CommonTypes.IOSampleType = None,
-        features: list[Feature] = None,
+        features: Optional[list[Feature]] = None,
         **kwargs,
     ):
         """
@@ -363,8 +363,8 @@ class ModelHandler(ABC, Generic[CommonTypes.ModelType, CommonTypes.IOSampleType]
 
     def set_labels(
         self,
-        to_add: dict[str, Union[str, int, float]] = None,
-        to_remove: list[str] = None,
+        to_add: Optional[dict[str, Union[str, int, float]]] = None,
+        to_remove: Optional[list[str]] = None,
     ):
         """
         Update the labels dictionary of this model artifact.
@@ -383,8 +383,8 @@ class ModelHandler(ABC, Generic[CommonTypes.ModelType, CommonTypes.IOSampleType]
 
     def set_parameters(
         self,
-        to_add: dict[str, Union[str, int, float]] = None,
-        to_remove: list[str] = None,
+        to_add: Optional[dict[str, Union[str, int, float]]] = None,
+        to_remove: Optional[list[str]] = None,
     ):
         """
         Update the parameters dictionary of this model artifact.
@@ -403,8 +403,8 @@ class ModelHandler(ABC, Generic[CommonTypes.ModelType, CommonTypes.IOSampleType]
 
     def set_metrics(
         self,
-        to_add: dict[str, CommonTypes.ExtraDataType] = None,
-        to_remove: list[str] = None,
+        to_add: Optional[dict[str, CommonTypes.ExtraDataType]] = None,
+        to_remove: Optional[list[str]] = None,
     ):
         """
         Update the metrics dictionary of this model artifact.
@@ -423,8 +423,8 @@ class ModelHandler(ABC, Generic[CommonTypes.ModelType, CommonTypes.IOSampleType]
 
     def set_extra_data(
         self,
-        to_add: dict[str, CommonTypes.ExtraDataType] = None,
-        to_remove: list[str] = None,
+        to_add: Optional[dict[str, CommonTypes.ExtraDataType]] = None,
+        to_remove: Optional[list[str]] = None,
     ):
         """
         Update the extra data dictionary of this model artifact.
@@ -509,7 +509,9 @@ class ModelHandler(ABC, Generic[CommonTypes.ModelType, CommonTypes.IOSampleType]
             self._import_custom_objects()
 
     @abstractmethod
-    def to_onnx(self, model_name: str = None, optimize: bool = True, **kwargs):
+    def to_onnx(
+        self, model_name: Optional[str] = None, optimize: bool = True, **kwargs
+    ):
         """
         Convert the model in this handler to an ONNX model.
 
@@ -525,13 +527,13 @@ class ModelHandler(ABC, Generic[CommonTypes.ModelType, CommonTypes.IOSampleType]
     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, CommonTypes.ExtraDataType] = None,
+        labels: Optional[dict[str, Union[str, int, float]]] = None,
+        parameters: Optional[dict[str, Union[str, int, float]]] = None,
+        inputs: Optional[list[Feature]] = None,
+        outputs: Optional[list[Feature]] = None,
+        metrics: Optional[dict[str, Union[int, float]]] = None,
+        artifacts: Optional[dict[str, Artifact]] = None,
+        extra_data: Optional[dict[str, CommonTypes.ExtraDataType]] = None,
         **kwargs,
     ):
         """
@@ -630,13 +632,13 @@ class ModelHandler(ABC, Generic[CommonTypes.ModelType, CommonTypes.IOSampleType]
 
     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, CommonTypes.ExtraDataType] = None,
+        labels: Optional[dict[str, Union[str, int, float]]] = None,
+        parameters: Optional[dict[str, Union[str, int, float]]] = None,
+        inputs: Optional[list[Feature]] = None,
+        outputs: Optional[list[Feature]] = None,
+        metrics: Optional[dict[str, Union[int, float]]] = None,
+        artifacts: Optional[dict[str, Artifact]] = None,
+        extra_data: Optional[dict[str, CommonTypes.ExtraDataType]] = None,
         **kwargs,
     ):
         """

mlrun/frameworks/_common/producer.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 #
 
+from typing import Optional
+
 import mlrun
 from mlrun.artifacts import Artifact
 
@@ -28,7 +30,7 @@ class Producer:
     def __init__(
         self,
         context: mlrun.MLClientCtx = None,
-        plans: list[Plan] = None,
+        plans: Optional[list[Plan]] = None,
     ):
         """
         Initialize a producer with the given plans. The producer will log the produced artifacts using the given

mlrun/frameworks/_dl_common/__init__.py

@@ -11,7 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-#
-# flake8: noqa  - this is until we take care of the F401 violations with respect to __all__ & sphinx
+
 from .model_handler import DLModelHandler
 from .utils import DLTypes, DLUtils

mlrun/frameworks/_dl_common/loggers/__init__.py

@@ -11,8 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-#
-# flake8: noqa  - this is until we take care of the F401 violations with respect to __all__ & sphinx
+
 from .logger import Logger
 from .mlrun_logger import MLRunLogger
 from .tensorboard_logger import TensorboardLogger