Skip to content

Update scikit-learn to version 1.5 #967

Open
tomerbv wants to merge 14 commits intomlrun:developmentfrom
tomerbv:scikitlearn_version
Open

Update scikit-learn to version 1.5 #967
tomerbv wants to merge 14 commits intomlrun:developmentfrom
tomerbv:scikitlearn_version

Conversation

@tomerbv
Copy link

@tomerbv tomerbv commented Jan 19, 2026

Updates scikit-learn dependency from 1.3 to ~=1.5 with necessary compatibility fixes.

Changes by function:

sklearn_classifier

  • Replaced deprecated MLRun utilities (get_sample, get_splits, gen_sklearn_model, eval_model_v2) with local
    implementations
  • Fixed test artifact paths for local mode execution

auto_trainer

  • Added monkey-patches for sklearn metrics to handle removed squared parameter (MLRun 1.10.0 compatibility)
  • Updated tests to use get_or_create_project() instead of new_project()

gen_class_data

  • Fixed test artifact path for local mode execution

describe

  • Updated dependencies only

All functions:

  • Updated requirements.txt: scikit-learn 1.3 → ~1.5
  • Updated item.yaml and regenerated function.yaml

fixes and patches for new scikit-learn version
changes in item.yaml and regenerate function.yaml
fix _get_dataframe to handle list/dict before accessing artifact_url
added feature name preservation logic in predict function
mlrun v 1.10
added skip for test_train in test_auto_trainer.py
@gilad-shaham gilad-shaham requested a review from guy1992l January 30, 2026 10:58
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant