Skip to content

sklearn_wrap.config

Pydantic-based YAML configuration for scikit-learn estimators.

Classes

Name Description
UntrustedModuleError Raised when a dotted path references a module not in the trusted allowlist.
EstimatorConfig Configuration for a scikit-learn compatible estimator.

Functions

Name Description
get_config Return a copy of the current sklearn-wrap configuration.
set_config Persistently update the sklearn-wrap configuration for the current thread.
reset_config Restore the sklearn-wrap configuration to its defaults for the current thread.
config_context Context manager to temporarily set sklearn-wrap configuration.