From 65d9cfa7dccb8085a2a39e838faca847e35da4e6 Mon Sep 17 00:00:00 2001
From: Czar Echavez <Czar.Echavez@envitia.com>
Date: Thu, 13 Jun 2024 15:11:34 +0100
Subject: [PATCH] #2658: I cast uncast on NIC space

---
 src/primaite/game/agent/observations/nic_observations.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/primaite/game/agent/observations/nic_observations.py b/src/primaite/game/agent/observations/nic_observations.py
index 125ec951..d6744cba 100644
--- a/src/primaite/game/agent/observations/nic_observations.py
+++ b/src/primaite/game/agent/observations/nic_observations.py
@@ -201,7 +201,7 @@ class NICObservation(AbstractObservation, identifier="NETWORK_INTERFACE"):
                             {"inbound": spaces.Discrete(11), "outbound": spaces.Discrete(11)}
                         )
 
-        return spaces.Dict(space)
+        return space
 
     @classmethod
     def from_config(cls, config: ConfigSchema, parent_where: WhereType = []) -> NICObservation: