mypy

QuTech-Delft · Dec 19, 2024 · 61213be · 61213be
1 parent b514431
commit 61213be
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -39,7 +39,6 @@ dependencies = [
     "gymnasium>=0.28.1",
     "networkx[default]>=2.6.3",
     "numpy>=1.21.6",
-    "numpy.typing",
     "pygame",
     "qiskit>=1.0.0",
 ]

diff --git a/qgym/envs/routing/routing_state.py b/qgym/envs/routing/routing_state.py
@@ -283,8 +283,9 @@ def obtain_observation(
             observation["connection_graph"] = self.connection_matrix
 
         if self.observe_legal_surpasses:
-            is_legal_surpass = np.array(
-                [self.is_legal_surpass(*gate) for gate in interaction_gates_ahead]
+            is_legal_surpass = np.fromiter(
+                (self.is_legal_surpass(*gate) for gate in interaction_gates_ahead),
+                dtype=np.bool,
             )
             observation["is_legal_surpass"] = is_legal_surpass