From cefd74f9037a879d512ad8311bc4d3ef3e4824c2 Mon Sep 17 00:00:00 2001 From: Ben Date: Mon, 4 Apr 2022 15:29:51 +0300 Subject: [PATCH] [Feature Store] Make _set_task_params public (#1863) --- mlrun/feature_store/api.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mlrun/feature_store/api.py b/mlrun/feature_store/api.py index a0a1e54c3a5..094c1e65319 100644 --- a/mlrun/feature_store/api.py +++ b/mlrun/feature_store/api.py @@ -360,7 +360,7 @@ def ingest( featureset, mlrun.api.schemas.AuthorizationAction.update ) run_config = run_config.copy() if run_config else RunConfig() - source, run_config.parameters = _set_task_params( + source, run_config.parameters = set_task_params( featureset, source, targets, run_config.parameters, infer_options, overwrite ) name = f"{featureset.metadata.name}_ingest" @@ -624,7 +624,7 @@ def _run_ingestion_job( featureset = get_feature_set_by_uri(featureset) run_config = run_config.copy() if run_config else RunConfig() - source, run_config.parameters = _set_task_params( + source, run_config.parameters = set_task_params( featureset, source, targets, run_config.parameters, infer_options ) @@ -677,7 +677,7 @@ def deploy_ingestion_service( kind=source.kind, name=featureset.metadata.name, ) - source, run_config.parameters = _set_task_params( + source, run_config.parameters = set_task_params( featureset, source, targets, run_config.parameters ) @@ -862,7 +862,7 @@ def _infer_from_static_df( return df -def _set_task_params( +def set_task_params( featureset: FeatureSet, source: DataSource = None, targets: List[DataTargetBase] = None,