Fix mypy ignores

invrs-io · Sep 25, 2023 · c203152 · c203152
1 parent ecdf131
commit c203152
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/invrs_gym/challenge/ceviche/challenge.py b/src/invrs_gym/challenge/ceviche/challenge.py
@@ -8,7 +8,7 @@
 import jax
 import jax.numpy as jnp
 import numpy as onp
-from totypes import types  # type: ignore[import]
+from totypes import types  # type: ignore[import,attr-defined,unused-ignore]
 
 from invrs_gym.challenge.ceviche import defaults
 from invrs_gym.loss import transmission_loss

diff --git a/src/invrs_gym/challenge/ceviche/defaults.py b/src/invrs_gym/challenge/ceviche/defaults.py
@@ -3,10 +3,15 @@
 from typing import Union
 
 import jax.numpy as jnp
-from ceviche_challenges import beam_splitter, mode_converter, model_base, params  # type: ignore[import]
+from ceviche_challenges import (  # type: ignore[import]
+    beam_splitter,
+    mode_converter,
+    model_base,
+    params,
+)
 from ceviche_challenges import units as u
 from ceviche_challenges import waveguide_bend, wdm
-from totypes import symmetry  # type: ignore[import]
+from totypes import symmetry  # type: ignore[import,attr-defined,unused-ignore]
 
 DeviceSpec = Union[
     beam_splitter.spec.BeamSplitterSpec,