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

lastFile:src/skore/project.py

src/skore/project.py

@@ -1,6 +1,8 @@
 """"""
 
+import abc
 import json
+from abc import abstractmethod
 from pathlib import Path
 from typing import Any, List
 
@@ -9,22 +11,23 @@ def get_filepath_from_project_name(project_name) -> Path:
     return None
 
 
-class Item:
+class Item(abc.ABC):
     """An item in the project."""
 
     @property
-    def raw():
-        ...
+    @abstractmethod
+    def raw(): ...
 
     @property
-    def transformed():
-        ...
+    @abstractmethod
+    def transformed(): ...
 
-    def to_json() -> str:
-        ...
+    @abstractmethod
+    def to_json() -> str: ...
+
+    @abstractmethod
+    def from_json(string: str): ...
 
-    def from_json(string:str):
-        ...
 
 class JSONItem(Item):
     """An item that is a JSON serializable Python object."""
@@ -34,44 +37,45 @@ def __init__(self, value):
 
     # to do cache
     @property
-    def raw():
+    def raw(self):
         return self.raw
 
     # to do cache
     @property
-    def transformed():
+    def transformed(self):
         return json.dumps(self.raw)
 
 
+
 class DataFrameItem(Item):
     """An item that is a pandas or polars data frame."""
+
     def __init__(self, value):
-        self.raw = value
+        self.__transformed = value.to_json(orient="split")
 
     @property
-    def raw():
-        raise NotImplementedError
+    def raw(self):...
+        # return self.__raw
 
     @property
-    def transformed():
-        return json.dumps(self.raw)
+    def transformed(self):
+        return self.__transformed
+
+    def from_json(string): ...
+
+    def to_json(string): ...
 
 
 class NumpyArrayItem(Item):
     @property
-    def raw():
-        ...
+    def raw(self): ...
 
     @property
-    def transformed():
-        ...
-
-    def to_json():
-        ...
+    def transformed(self): ...
 
-    def from_json():
-        ...
+    def to_json(self): ...
 
+    def from_json(self): ...
 
 
 class MediaItem(Item):
@@ -84,10 +88,6 @@ class ScikitLearnModelItem(Item):
     def __init__(self, value):
         self.value = value
 
-    @staticmethod
-    def transform(value):
-        return html
-
 
 class Transformer:
     @staticmethod

tests/test_dataframe_item.py

@@ -0,0 +1,10 @@
+import pandas as pd
+from skore.project import DataFrameItem
+
+
+def test_dataframe_item():
+    df = pd.DataFrame([[1, 2, 3], [4, 5, 6]])
+    item = DataFrameItem(df)
+
+    assert item.raw == df
+    assert item.transformed == None