Fix mypy CI failure: use correct type: ignore error codes for rearrange fallback

The try/except ImportError fallback for rearrange triggers different mypy
error codes across versions (misc vs no-redef). Suppress both to ensure
CI passes consistently.

Author: jonwiggins

src/visualization/attention_viz.py

@@ -20,7 +20,7 @@
     from einops import rearrange
 except ImportError:
 
-    def rearrange(  # type: ignore[misc]
+    def rearrange(  # type: ignore[misc, no-redef]
         tensor: torch.Tensor | list[torch.Tensor], pattern: str, **axes_lengths: Any
     ) -> torch.Tensor:
         """Fallback rearrange for basic patterns."""

src/visualization/masking_viz.py

@@ -19,7 +19,7 @@
     from einops import rearrange
 except ImportError:
 
-    def rearrange(  # type: ignore[misc]
+    def rearrange(  # type: ignore[misc, no-redef]
         tensor: torch.Tensor | list[torch.Tensor], pattern: str, **axes_lengths: Any
     ) -> torch.Tensor:
         """Fallback rearrange for basic patterns."""