use get instead of dict key

probabl-ai · Dec 19, 2024 · 96c751a · 96c751a
1 parent 659a86b
commit 96c751a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/skore/src/skore/item/item.py b/skore/src/skore/item/item.py
@@ -85,7 +85,6 @@ def __repr__(self) -> str:
         return f"{self.__class__.__name__}(...)"
 
     def _repr_mimebundle_(self, include=None, exclude=None):
-        del include, exclude
         return {"text/html": self._repr_html_()}
 
     def _repr_html_(self):

diff --git a/skore/src/skore/ui/project_routes.py b/skore/src/skore/ui/project_routes.py
@@ -39,10 +39,10 @@ def __item_as_serializable(name: str, item: Item) -> SerializableItem:
     d = item.as_serializable_dict()
     return SerializableItem(
         name=name,
-        media_type=d["media_type"],
-        value=d["value"],
-        updated_at=d["updated_at"],
-        created_at=d["created_at"],
+        media_type=d.get("media_type"),
+        value=d.get("value"),
+        updated_at=d.get("updated_at"),
+        created_at=d.get("created_at"),
     )