assert not none in imagesets

Author: jwmueller

src/cleanvision/imagelab.py

@@ -501,7 +501,9 @@ def _visualize(
 
             scores = sorted_df.head(num_images)[get_score_colname(issue_type)]
             indices = scores.index.tolist()
-            images: list[Image.Image] = [self._dataset[i] for i in indices]
+            images = [self._dataset[i] for i in indices]
+            for image in images:
+                assert isinstance(image, Image.Image), "image must be of type Image.Image"
 
             # construct title info
             title_info = {"scores": [f"score : {x:.4f}" for x in scores]}
@@ -526,7 +528,9 @@ def _visualize(
             image_sets = []
             for indices in image_sets_indices:
                 image_sets.append([self._dataset[index] for index in indices])
-            image_sets: list[list[Image.Image]] = image_sets
+            for image_set in image_sets:
+                for img in image_set:
+                    assert isinstance(img, Image.Image), "image must be of type Image.Image"
 
             title_info_sets = []
             for s in image_sets_indices: