mob next [ci-skip] [ci skip] [skip ci]

lastFile:tests/unit/test_project.py

src/skore/project.py

@@ -61,16 +61,20 @@ def untransform(serialized: str, raw_class_name: str) -> Item:
             raw = json.loads(serialized)
         case "pandas.DataFrame":
             import pandas
+
             raw = pandas.read_json(serialized, orient="split")
         case "numpy.ndarray":
             import numpy
+
             raw = numpy.array(json.loads(serialized))
+        case "sklearn.base.BaseEstimator":
+            import skops.io
+
+            o = json.loads(serialized)
+            unserialized = base64.b64decode(o["skops"])
+            raw = skops.io.loads(unserialized)
 
-    return Item(
-        raw=raw,
-        raw_class_name=raw_class_name,
-        serialized=serialized
-    )
+    return Item(raw=raw, raw_class_name=raw_class_name, serialized=serialized)
 
 
 class Project:

tests/unit/test_project.py

@@ -1,6 +1,7 @@
 import json
 
 import numpy
+import numpy.testing
 import pandas
 import pandas.testing
 import sklearn.svm
@@ -76,10 +77,7 @@ def test_untransform_pandas_dataframe():
         raw_class_name="pandas.DataFrame",
     )
 
-    pandas.testing.assert_frame_equal(
-        item.raw,
-        o
-    )
+    pandas.testing.assert_frame_equal(item.raw, o)
 
 
 def test_untransform_numpy_ndarray():
@@ -89,4 +87,12 @@ def test_untransform_numpy_ndarray():
         raw_class_name="numpy.ndarray",
     )
 
-    assert o == item.raw
+    numpy.testing.assert_array_equal(o, item.raw)
+
+
+def test_untransform_sklearn_model():
+    o = sklearn.svm.SVC()
+    t = transform(o)
+    u = untransform(t.serialized, t.raw_class_name)
+
+    assert isinstance(u.raw, sklearn.svm.SVC)