more casts

jwmueller · web-flow · commit 4d3049e2cd28 · 2025-12-17T02:03:53.000-08:00
diff --git a/src/cleanvision/imagelab.py b/src/cleanvision/imagelab.py
@@ -632,6 +632,7 @@ def visualize(
             )
         elif indices:
             images = [self._dataset[i] for i in indices]
+            images = cast(list[Image.Image], images)
             title_info = {"name": [self._dataset.get_name(i) for i in indices]}
             VizManager.individual_images(
                 images,
@@ -647,6 +648,7 @@ def visualize(
                     self._dataset.index, min(num_images, len(self._dataset))
                 )
                 images = [self._dataset[i] for i in image_indices]
+                images = cast(list[Image.Image], images)
                 title_info = {
                     "name": [self._dataset.get_name(i) for i in image_indices]
                 }

Original file line number	Diff line number	Diff line change
`@@ -632,6 +632,7 @@ def visualize(`
`632`	`632`	`)`
`633`	`633`	`elif indices:`
`634`	`634`	`images = [self._dataset[i] for i in indices]`
	`635`	`+ images = cast(list[Image.Image], images)`
`635`	`636`	`title_info = {"name": [self._dataset.get_name(i) for i in indices]}`
`636`	`637`	`VizManager.individual_images(`
`637`	`638`	`images,`
`@@ -647,6 +648,7 @@ def visualize(`
`647`	`648`	`self._dataset.index, min(num_images, len(self._dataset))`
`648`	`649`	`)`
`649`	`650`	`images = [self._dataset[i] for i in image_indices]`
	`651`	`+ images = cast(list[Image.Image], images)`
`650`	`652`	`title_info = {`
`651`	`653`	`"name": [self._dataset.get_name(i) for i in image_indices]`
`652`	`654`	`}`