(DiamondLightSource/hyperion#1169) Fix issues with mypy

DiamondLightSource · Apr 12, 2024 · 8881620 · 8881620
1 parent 72dd195
commit 8881620
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/dodal/devices/oav/grid_overlay.py b/src/dodal/devices/oav/grid_overlay.py
@@ -15,7 +15,7 @@ class Orientation(Enum):
 
 
 def _add_parallel_lines_to_image(
-    image: Image,
+    image: Image.Image,
     start_x: int,
     start_y: int,
     line_length: int,

diff --git a/tests/devices/unit_tests/test_oav.py b/tests/devices/unit_tests/test_oav.py
@@ -68,7 +68,7 @@ def test_snapshot_trigger_saves_to_correct_file(
 ):
     image = PIL.Image.open("test")
     mock_save = MagicMock()
-    image.save = mock_save
+    image.save = mock_save  # type: ignore
     mock_open.return_value.__enter__.return_value = image
     st = fake_oav.snapshot.trigger()
     st.wait()