Fix codestyle

Author: astrofrog

jupyter_output_monitor/_monitor.py

@@ -53,7 +53,10 @@ def monitor_group():
 @click.option(
     "--atol",
     default=0,
-    help="If an output image for a cell exists, a new image will only be written out if the maximum uint8 difference between the two exceeds atol",
+    help=(
+        "If an output image for a cell exists, a new image will only be written "
+        "out if the maximum uint8 difference between the two exceeds atol"
+    ),
 )
 @click.option("--headless", is_flag=True, help="Whether to run in headless mode")
 def monitor(notebook, url, output, wait_after_execute, atol, headless):

jupyter_output_monitor/_utils.py

@@ -47,7 +47,7 @@ def max_uint8_difference(image1_bytes, image2_bytes):
 
     # Ensure both images have the same dimensions
     if array1.shape != array2.shape:
-        raise ValueError("Images have different dimensions")
+        return 256
 
     # Calculate the absolute difference
     diff = np.abs(array1.astype(np.int16) - array2.astype(np.int16))

jupyter_output_monitor/tests/test_monitor.py

@@ -26,8 +26,8 @@ def test_simple(output_path, threshold):
             "--output",
             str(output_path),
             "--headless",
-        ]
-        + extra,
+            *extra,
+        ],
         check=True,
     )
 
@@ -38,7 +38,7 @@ def test_simple(output_path, threshold):
 
     # Output screenshots
     if threshold:
-        assert len(list(output_path.glob("output-*.png"))) in (4,5)
+        assert len(list(output_path.glob("output-*.png"))) in (4, 5)
     else:
         assert len(list(output_path.glob("output-*.png"))) >= 4
 
@@ -50,11 +50,10 @@ def test_simple(output_path, threshold):
 
     # Specifically for cell with index 33
     if threshold:
-        assert len(list(output_path.glob("output-033-*.png"))) in (3,4)
+        assert len(list(output_path.glob("output-033-*.png"))) in (3, 4)
     else:
         assert len(list(output_path.glob("output-033-*.png"))) >= 3
 
-
     # Check that event log exists and is parsable
     with open(output_path / "event_log.csv") as f:
         reader = csv.reader(f, delimiter=",")

pyproject.toml

@@ -50,6 +50,7 @@ lint.ignore = [
     "D103",
     "D104",
     "C901",
+    "PLR0913",
     "PLR0915",
     "PLR2004",
     "DTZ",