mlrun 1.7.2__py3-none-any.whl → 1.8.0rc1__py3-none-any.whl

mlrun/feature_store/api.py

@@ -99,21 +99,21 @@ def _features_to_vector_and_check_permissions(features, update_stats):
 def get_offline_features(
     feature_vector: Union[str, FeatureVector],
     entity_rows=None,
-    entity_timestamp_column: str = None,
+    entity_timestamp_column: 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: Optional[list[str]] = None,
+    start_time: Optional[Union[str, datetime]] = None,
+    end_time: 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: Optional[str] = None,
+    engine_args: Optional[dict] = None,
+    query: Optional[str] = None,
+    order_by: Optional[Union[str, list[str]]] = None,
+    spark_service: Optional[str] = None,
+    timestamp_for_filtering: Optional[Union[str, dict[str, str]]] = None,
+    additional_filters: Optional[list] = None,
 ):
     """retrieve offline feature vector results
 
@@ -209,20 +209,20 @@ def get_offline_features(
 def _get_offline_features(
     feature_vector: Union[str, FeatureVector],
     entity_rows=None,
-    entity_timestamp_column: str = None,
+    entity_timestamp_column: 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: Optional[list[str]] = None,
+    start_time: Optional[Union[str, datetime]] = None,
+    end_time: 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,
+    engine: Optional[str] = None,
+    engine_args: Optional[dict] = None,
+    query: Optional[str] = None,
+    order_by: Optional[Union[str, list[str]]] = None,
+    spark_service: Optional[str] = None,
+    timestamp_for_filtering: Optional[Union[str, dict[str, str]]] = None,
     additional_filters=None,
 ) -> Union[OfflineVectorResponse, RemoteVectorResponse]:
     if entity_rows is None and entity_timestamp_column is not None:
@@ -297,9 +297,9 @@ def get_online_feature_service(
     feature_vector: Union[str, FeatureVector],
     run_config: RunConfig = None,
     fixed_window_type: FixedWindowType = FixedWindowType.LastClosedWindow,
-    impute_policy: dict = None,
+    impute_policy: Optional[dict] = None,
     update_stats: bool = False,
-    entity_keys: list[str] = None,
+    entity_keys: Optional[list[str]] = None,
 ):
     """initialize and return online feature vector service api,
     returns :py:class:`~mlrun.feature_store.OnlineVectorService`
@@ -378,9 +378,9 @@ def _get_online_feature_service(
     feature_vector: Union[str, FeatureVector],
     run_config: RunConfig = None,
     fixed_window_type: FixedWindowType = FixedWindowType.LastClosedWindow,
-    impute_policy: dict = None,
+    impute_policy: Optional[dict] = None,
     update_stats: bool = False,
-    entity_keys: list[str] = None,
+    entity_keys: Optional[list[str]] = None,
 ) -> OnlineVectorService:
     if isinstance(feature_vector, FeatureVector):
         update_stats = True
@@ -450,7 +450,7 @@ def _get_namespace(run_config: RunConfig) -> dict[str, Any]:
 def ingest(
     featureset: Union[FeatureSet, str] = None,
     source=None,
-    targets: list[DataTargetBase] = None,
+    targets: Optional[list[DataTargetBase]] = None,
     namespace=None,
     return_df: bool = True,
     infer_options: InferOptions = InferOptions.default(),
@@ -530,7 +530,7 @@ def ingest(
 def _ingest(
     featureset: Union[FeatureSet, str] = None,
     source=None,
-    targets: list[DataTargetBase] = None,
+    targets: Optional[list[DataTargetBase]] = None,
     namespace=None,
     return_df: bool = True,
     infer_options: InferOptions = InferOptions.default(),
@@ -783,11 +783,11 @@ def _ingest(
 def preview(
     featureset: FeatureSet,
     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
 
@@ -825,11 +825,11 @@ def preview(
 def _preview(
     featureset: FeatureSet,
     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:
     if isinstance(source, pd.DataFrame):
         source = _rename_source_dataframe_columns(source)
@@ -895,8 +895,8 @@ def _preview(
 def _run_ingestion_job(
     featureset: Union[FeatureSet, str],
     source: DataSource = None,
-    targets: list[DataTargetBase] = None,
-    name: str = None,
+    targets: Optional[list[DataTargetBase]] = None,
+    name: Optional[str] = None,
     infer_options: InferOptions = InferOptions.default(),
     run_config: RunConfig = None,
 ):
@@ -920,8 +920,8 @@ def _run_ingestion_job(
 def deploy_ingestion_service_v2(
     featureset: Union[FeatureSet, str],
     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]:
@@ -963,8 +963,8 @@ def deploy_ingestion_service_v2(
 def _deploy_ingestion_service_v2(
     featureset: Union[FeatureSet, str],
     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]:
@@ -1026,7 +1026,7 @@ def _ingest_with_spark(
     spark=None,
     featureset: Union[FeatureSet, str] = None,
     source: BaseSourceDriver = None,
-    targets: list[BaseStoreTarget] = None,
+    targets: Optional[list[BaseStoreTarget]] = None,
     infer_options: InferOptions = InferOptions.default(),
     mlrun_context=None,
     namespace=None,
@@ -1199,8 +1199,8 @@ def _infer_from_static_df(
 def set_task_params(
     featureset: FeatureSet,
     source: DataSource = None,
-    targets: list[DataTargetBase] = None,
-    parameters: dict = None,
+    targets: Optional[list[DataTargetBase]] = None,
+    parameters: Optional[dict] = None,
     infer_options: InferOptions = InferOptions.Null,
     overwrite=None,
 ):

mlrun/feature_store/common.py

@@ -178,17 +178,17 @@ class RunConfig:
     def __init__(
         self,
         function: typing.Union[str, FunctionReference, BaseRuntime] = None,
-        local: bool = None,
-        image: str = None,
-        kind: str = None,
-        handler: str = None,
-        parameters: dict = None,
-        watch: bool = None,
+        local: typing.Optional[bool] = None,
+        image: typing.Optional[str] = None,
+        kind: typing.Optional[str] = None,
+        handler: typing.Optional[str] = None,
+        parameters: typing.Optional[dict] = None,
+        watch: typing.Optional[bool] = None,
         owner=None,
         credentials: typing.Optional[mlrun.model.Credentials] = None,
-        code: str = None,
-        requirements: typing.Union[str, list[str]] = None,
-        extra_spec: dict = None,
+        code: typing.Optional[str] = None,
+        requirements: typing.Optional[typing.Union[str, list[str]]] = None,
+        extra_spec: typing.Optional[dict] = None,
         auth_info=None,
     ):
         """class for holding function and run specs for jobs and serving functions

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.