[BUG] Fix Sklearn Models detection by safely importing openml-sklearn

[EmanAbdelhaleem]

Metadata

• Reference Issue: Fixes [BUG] Fix Sklearn Models detection by safely registering SklearnExtension #1542

Details

Fixed sklearn models detection by safely importing openml-sklearn at openml/runs/__init__.py

[geetu040]

Nice, the safe import logic looks good.

Since this is specific to models and flows I would suggest we move it inside both functions in extensions/functions.py as you suggested before. Though it's still not the best place to put this, but we are restricted by circular imports and can improve the logic afterwards when extension classes are refactored

Also, could you improve the SKLEARN_HINT message, you can suggest to install it using pip install openml-sklearn

[fkiraly]

Quick question - why does this work, given that openml_sklearn is not actually directly used? Is there some background magic that does something non-explicit, in case the module gets imported?

That feels very dangerous design, if it were so (fully aware that it is not yours).

[EmanAbdelhaleem]

Is there some background magic that does something non-explicit, in case the module gets imported?

Yes, the SklearnExtension gets registered, the issue was that it wasn't registered, and actually with the current design, it only gets registered when openml-sklearn is imported.
Check here: https://github.com/openml/openml-sklearn/blob/main/src/openml_sklearn/__init__.py#L15C1-L16C1

[codecov-commenter]

Codecov Report

❌ Patch coverage is 60.00000% with 2 lines in your changes missing coverage. Please review.
✅ Project coverage is 52.71%. Comparing base (bd8ae77) to head (92091c1).
⚠️ Report is 1 commits behind head on main.

Files with missing lines	Patch %	Lines
openml/extensions/functions.py	60.00%	2 Missing ⚠️

Additional details and impacted files

@@             Coverage Diff             @@
##             main    #1556       +/-   ##
===========================================
- Coverage   79.14%   52.71%   -26.43%     
===========================================
  Files          36       36               
  Lines        4320     4325        +5     
===========================================
- Hits         3419     2280     -1139     
- Misses        901     2045     +1144     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.