Skip to content

Conversation

@EmanAbdelhaleem
Copy link

Metadata

Details

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

Copy link

@geetu040 geetu040 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

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

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.

[BUG] Fix Sklearn Models detection by safely registering SklearnExtension

2 participants