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

lastFile:tests/unit/test_project.py

src/skore/project.py

@@ -1,5 +1,6 @@
 """"""
 
+import base64
 import json
 from dataclasses import dataclass
 from typing import Any, List
@@ -14,8 +15,7 @@ class Item:
     raw_class_name: str | None = None
 
 
-
-def transform(o: Any) -> Item:
+def transform(o: Any) -> Item | None:
     """Transform an object into an item."""
     try:
         serialized = json.dumps(o)
@@ -44,11 +44,12 @@ def transform(o: Any) -> Item:
                 raw_class_name="sklearn.base.BaseEstimator",
                 serialized=json.dumps(
                     {
-                        "skops": skops.io.dumps(o).decode(),
+                        "skops": base64.b64encode(skops.io.dumps(o)).decode("ascii"),
                         "html": sklearn.utils.estimator_html_repr(o),
                     }
                 ),
             )
+        return None
 
 
 class Project:

tests/unit/test_project.py

@@ -1,8 +1,8 @@
 import json
-import pandas
+
 import numpy
+import pandas
 import sklearn.svm
-import skops.io
 from skore.project import Item, transform
 
 
@@ -37,16 +37,19 @@ def test_transform_numpy_ndarray():
     assert actual == expected
 
 
-def test_transform_sklearn_base_baseestimator():
+def test_transform_sklearn_base_baseestimator(monkeypatch):
+    monkeypatch.setattr("sklearn.utils.estimator_html_repr", lambda _: "")
+    monkeypatch.setattr("skops.io.dumps", lambda _: "")
+
     o = sklearn.svm.SVC()
     actual = transform(o)
     expected = Item(
         raw=o,
         raw_class_name="numpy.ndarray",
         serialized=json.dumps(
             {
-                "skops": skops.io.dumps(o).decode(),
-                "html": sklearn.utils.estimator_html_repr(o),
+                "skops": "",
+                "html": "",
             }
         ),
     )