From dcf411a8273d664089a7368db195f7346b18dc9d Mon Sep 17 00:00:00 2001
From: William Blum <william.blum@microsoft.com>
Date: Tue, 6 Aug 2024 23:33:46 +0000
Subject: [PATCH] .

---
 cyberbattle/_env/cyberbattle_env.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cyberbattle/_env/cyberbattle_env.py b/cyberbattle/_env/cyberbattle_env.py
index 0b33b9a..43687f7 100644
--- a/cyberbattle/_env/cyberbattle_env.py
+++ b/cyberbattle/_env/cyberbattle_env.py
@@ -321,8 +321,8 @@ def __init__(self, bounds: EnvironmentBounds):
 
 
 class CyberBattleSpaceKind(Env[Observation, Action]):
-    action_space: DiscriminatedUnion
-    observation_space: ObservationSpaceType
+    action_space: DiscriminatedUnion # type: ignore
+    observation_space: ObservationSpaceType # type: ignore
 
 
 class CyberBattleEnv(CyberBattleSpaceKind):