ENH All sklearn estimators in trusted list

skops/io/_general.py

@@ -8,7 +8,7 @@
 import numpy as np
 
 from ._audit import Node, get_tree
-from ._trusted_types import PRIMITIVE_TYPE_NAMES
+from ._trusted_types import PRIMITIVE_TYPE_NAMES, SKLEARN_ESTIMATOR_TYPE_NAMES
 from ._utils import (
     LoadContext,
     SaveContext,
@@ -383,7 +383,7 @@ def __init__(
 
         self.children = {"attrs": attrs}
         # TODO: what do we trust?
-        self.trusted = self._get_trusted(trusted, [])
+        self.trusted = self._get_trusted(trusted, default=SKLEARN_ESTIMATOR_TYPE_NAMES)
 
     def _construct(self):
         cls = gettype(self.module_name, self.class_name)

skops/io/_trusted_types.py

@@ -1,3 +1,13 @@
+from sklearn.utils import all_estimators
+
+from ._utils import get_type_name
+
 PRIMITIVES_TYPES = [int, float, str, bool]
 
 PRIMITIVE_TYPE_NAMES = ["builtins." + t.__name__ for t in PRIMITIVES_TYPES]
+
+SKLEARN_ESTIMATOR_TYPE_NAMES = [
+    get_type_name(estimator_class)
+    for _, estimator_class in all_estimators()
+    if get_type_name(estimator_class).startswith("sklearn.")
+]

skops/io/tests/test_audit.py

@@ -162,12 +162,4 @@ def test_complex_pipeline_untrusted_set():
 
     untrusted = get_untrusted_types(data=dumps(clf))
     type_names = [x.split(".")[-1] for x in untrusted]
-    assert type_names == [
-        "sqrt",
-        "square",
-        "LogisticRegression",
-        "FeatureUnion",
-        "Pipeline",
-        "StandardScaler",
-        "FunctionTransformer",
-    ]
+    assert type_names == ["sqrt", "square"]

skops/io/tests/test_persist.py

@@ -468,7 +468,7 @@ def get_input(estimator):
 @pytest.mark.parametrize(
     "estimator", _tested_estimators(), ids=_get_check_estimator_ids
 )
-def test_can_persist_fitted(estimator, request):
+def test_can_persist_fitted(estimator):
     """Check that fitted estimators can be persisted and return the right results."""
     set_random_state(estimator, random_state=0)
 
@@ -491,6 +491,12 @@ def test_can_persist_fitted(estimator, request):
     loaded = loads(dumped, trusted=untrusted_types)
     assert_params_equal(estimator.__dict__, loaded.__dict__)
 
+    # test that most sklearn estimators are not in untrusted_types
+    sklearn_untrusted_types = [
+        type_ for type_ in untrusted_types if type_.startswith("sklearn.")
+    ]
+    assert len(sklearn_untrusted_types) == 0
+
     for method in [
         "predict",
         "predict_proba",