Skip to content

[ENH] allow arbitrary base class in all_objects#284

Merged
fkiraly merged 1 commit into
mainfrom
all_objects-improved
Feb 9, 2024
Merged

[ENH] allow arbitrary base class in all_objects#284
fkiraly merged 1 commit into
mainfrom
all_objects-improved

Conversation

@fkiraly
Copy link
Copy Markdown
Contributor

@fkiraly fkiraly commented Feb 4, 2024

This PR extends all_objects to allow an arbitrary base class - currently, it only lists classes inheriting from skbase all_objects. With the change in this PR, something like this will give the expected results - of all estimators in sklearn:

from skbase.lookup import all_objects
from sklearn.base import BaseEstimator

all_objects(BaseEstimator, package_name="sklearn", modules_to_ignore=["tests"], as_dataframe=True)

Minor changes also contained:

  • removes unused args exclude_estimators and ignore_modules

@fkiraly fkiraly added the enhancement Adding new functionality label Feb 4, 2024
@fkiraly fkiraly merged commit 5d1b7da into main Feb 9, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement Adding new functionality

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant