Checks for images using PIL only if available

src/distilabel/__init__.py

@@ -14,6 +14,6 @@
 
 from rich import traceback as rich_traceback
 
-__version__ = "1.5.2"
+__version__ = "1.6.0"
 
 rich_traceback.install(show_locals=True)

src/distilabel/distiset.py

@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import importlib.util
 import json
 import logging
 import os.path as posixpath
@@ -52,6 +53,19 @@
     from distilabel.pipeline._dag import DAG
 
 
+def is_PIL_available() -> bool:
+    """Checks if the PIL library is available.
+
+    Returns:
+        True if the PIL library is available, False otherwise.
+    """
+    try:
+        importlib.util.find_spec("PIL")
+    except ImportError:
+        return False
+    return True
+
+
 class Distiset(dict):
     """Convenient wrapper around `datasets.Dataset` to push to the Hugging Face Hub.
 
@@ -187,11 +201,14 @@ def _get_card(
             record = (
                 dataset[0] if not isinstance(dataset, dict) else dataset["train"][0]
             )
-            from PIL import ImageFile
+            if is_PIL_available():
+                from PIL import ImageFile
+            else:
+                ImageFile = None
 
             for key, value in record.items():
                 # If the value is an image, we set it to an empty string to avoid the `README.md` to huge
-                if isinstance(value, ImageFile.ImageFile):
+                if ImageFile and isinstance(value, ImageFile.ImageFile):
                     value = ""
                 # If list is too big, the `README.md` generated will be huge so we truncate it
                 elif isinstance(value, list):