Obviate type ignore statments

invrs-io · Nov 9, 2023 · f59ac88 · f59ac88
1 parent de58d33
commit f59ac88
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/invrs_gym/challenges/sorter/common.py b/src/invrs_gym/challenges/sorter/common.py
@@ -239,14 +239,12 @@ def response(
 
         spec = dataclasses.replace(
             self.spec,
-            thickness_cap=params[THICKNESS_CAP].array,  # type: ignore[arg-type]
-            thickness_metasurface=(
-                params[THICKNESS_METASURFACE].array  # type: ignore[arg-type]
-            ),
-            thickness_spacer=params[THICKNESS_SPACER].array,  # type: ignore[arg-type]
+            thickness_cap=jnp.asarray(params[THICKNESS_CAP].array),
+            thickness_metasurface=jnp.asarray(params[THICKNESS_METASURFACE].array),
+            thickness_spacer=jnp.asarray(params[THICKNESS_SPACER].array),
         )
         return simulate_sorter(
-            density_array=params[DENSITY_METASURFACE].array,  # type: ignore[arg-type]
+            density_array=jnp.asarray(params[DENSITY_METASURFACE].array),
             spec=spec,
             wavelength=jnp.asarray(wavelength),
             polar_angle=jnp.asarray(polar_angle),