Merge pull request #823 from aviaIguazio/development

[feature_selection] update function yaml
mlrun · Sep 12, 2024 · 8f3f226 · 8f3f226
2 parents 55e1dc0 + 75dbafd
commit 8f3f226
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 37 deletions.
diff --git a/feature_selection/function.yaml b/feature_selection/function.yaml
diff --git a/feature_selection/item.yaml b/feature_selection/item.yaml
@@ -12,14 +12,14 @@ labels:
   author: orz
 maintainers: []
 marketplaceType: ''
-mlrunVersion: 1.1.0
+mlrunVersion: 1.6.4
 name: feature-selection
-platformVersion: 3.5.0
+platformVersion: 3.6.0
 spec:
   filename: feature_selection.py
   handler: feature_selection
   image: mlrun/mlrun
   kind: job
   requirements: []
 url: ''
-version: 1.4.0
+version: 1.5.0
diff --git a/feature_selection/requirements.txt b/feature_selection/requirements.txt
@@ -1,4 +1,4 @@
-scikit-learn~=1.0.2
+scikit-learn
 matplotlib
 seaborn
 scikit-plot

diff --git a/feature_selection/test_feature_selection.py b/feature_selection/test_feature_selection.py
@@ -44,5 +44,5 @@ def test_run_local_feature_selection():
         inputs={'df_artifact': 'data/metrics.pq'},
         artifact_path='artifacts/',
     )
-    assert run.artifact('feature_scores').get() and run.artifact('selected_features').get()
+    assert run.outputs['feature_scores'] and run.outputs['selected_features']
     _delete_outputs({ARTIFACTS_PATH, RUNS_PATH, SCHEDULES_PATH})