From e3cbab12ff5882de01516c39a81755c055c1d5f0 Mon Sep 17 00:00:00 2001 From: Stan van der Linde Date: Tue, 7 Jan 2025 14:40:24 +0100 Subject: [PATCH] mypy --- qgym/envs/routing/routing_state.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qgym/envs/routing/routing_state.py b/qgym/envs/routing/routing_state.py index 20f2685..6133e97 100644 --- a/qgym/envs/routing/routing_state.py +++ b/qgym/envs/routing/routing_state.py @@ -291,7 +291,7 @@ def obtain_observation( is_legal_surpass = np.fromiter( iter=starmap(self.is_legal_surpass, interaction_gates_ahead), count=len(interaction_gates_ahead), - dtype=np.int8, + dtype=np.int_, ) observation["is_legal_surpass"] = is_legal_surpass