Replace third-part libraries imports by lazy imports (#350)

Fix missing imports in project.py by implementing a lazy isinstance
function.

---------

Co-authored-by: Auguste Baum <[email protected]>

Author: thomass-dev

src/skore/item/media_item.py

@@ -6,7 +6,7 @@
 from __future__ import annotations
 
 from io import BytesIO
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
 if TYPE_CHECKING:
     from altair.vegalite.v5.schema.core import TopLevelSpec as Altair
@@ -16,6 +16,13 @@
 from skore.item.item import Item
 
 
+def lazy_is_instance(object: Any, cls_fullname: str) -> bool:
+    """Return True if object is an instance of a class named `cls_fullname`."""
+    return cls_fullname in {
+        f"{cls.__module__}.{cls.__name__}" for cls in object.__class__.__mro__
+    }
+
+
 class MediaItem(Item):
     """
     A class to represent a media item.
@@ -76,19 +83,15 @@ def factory(cls, media, *args, **kwargs):
         MediaItem
             A new MediaItem instance.
         """
-        media_mro_fullnames = {
-            f"{cls.__module__}.{cls.__name__}" for cls in media.__class__.__mro__
-        }
-
-        if "builtins.bytes" in media_mro_fullnames:
+        if lazy_is_instance(media, "builtins.bytes"):
             return cls.factory_bytes(media, *args, **kwargs)
-        if "builtins.str" in media_mro_fullnames:
+        if lazy_is_instance(media, "builtins.str"):
             return cls.factory_str(media, *args, **kwargs)
-        if "altair.vegalite.v5.schema.core.TopLevelSpec" in media_mro_fullnames:
+        if lazy_is_instance(media, "altair.vegalite.v5.schema.core.TopLevelSpec"):
             return cls.factory_altair(media, *args, **kwargs)
-        if "matplotlib.figure.Figure" in media_mro_fullnames:
+        if lazy_is_instance(media, "matplotlib.figure.Figure"):
             return cls.factory_matplotlib(media, *args, **kwargs)
-        if "PIL.Image.Image" in media_mro_fullnames:
+        if lazy_is_instance(media, "PIL.Image.Image"):
             return cls.factory_pillow(media, *args, **kwargs)
 
         raise NotImplementedError(f"Type '{media.__class__}' is not yet supported")

src/skore/project.py

@@ -3,16 +3,9 @@
 from pathlib import Path
 from typing import Any
 
-import altair
-import matplotlib
-import numpy
-import pandas
-import PIL
-import sklearn
-
 from skore.item import Item
 from skore.item.item_repository import ItemRepository
-from skore.item.media_item import MediaItem
+from skore.item.media_item import MediaItem, lazy_is_instance
 from skore.item.numpy_array_item import NumpyArrayItem
 from skore.item.pandas_dataframe_item import PandasDataFrameItem
 from skore.item.primitive_item import PrimitiveItem, is_primitive
@@ -25,17 +18,17 @@ def object_to_item(o: Any) -> Item:
     """Transform an object into an Item."""
     if is_primitive(o):
         return PrimitiveItem.factory(o)
-    elif isinstance(o, pandas.DataFrame):
+    elif lazy_is_instance(o, "pandas.core.frame.DataFrame"):
         return PandasDataFrameItem.factory(o)
-    elif isinstance(o, numpy.ndarray):
+    elif lazy_is_instance(o, "numpy.ndarray"):
         return NumpyArrayItem.factory(o)
-    elif isinstance(o, sklearn.base.BaseEstimator):
+    elif lazy_is_instance(o, "sklearn.base.BaseEstimator"):
         return SklearnBaseEstimatorItem.factory(o)
-    elif isinstance(o, altair.vegalite.v5.schema.core.TopLevelSpec):
+    elif lazy_is_instance(o, "altair.vegalite.v5.schema.core.TopLevelSpec"):
         return MediaItem.factory_altair(o)
-    elif isinstance(o, matplotlib.figure.Figure):
+    elif lazy_is_instance(o, "matplotlib.figure.Figure"):
         return MediaItem.factory_matplotlib(o)
-    elif isinstance(o, PIL.Image.Image):
+    elif lazy_is_instance(o, "PIL.Image.Image"):
         return MediaItem.factory_pillow(o)
     else:
         raise NotImplementedError(f"Type {o.__class__.__name__} is not supported yet.")