imgsets typing

Author: jwmueller

src/cleanvision/imagelab.py

@@ -501,7 +501,7 @@ def _visualize(
 
             scores = sorted_df.head(num_images)[get_score_colname(issue_type)]
             indices = scores.index.tolist()
-            images = [self._dataset[i] for i in indices]
+            images: list[Image.Image] = [self._dataset[i] for i in indices]
 
             # construct title info
             title_info = {"scores": [f"score : {x:.4f}" for x in scores]}
@@ -526,6 +526,7 @@ 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
 
             title_info_sets = []
             for s in image_sets_indices: