#2238 - Fixed the observations issue causing tests to fail

src/primaite/simulator/network/hardware/base.py

@@ -123,6 +123,8 @@ class NetworkInterface(SimComponent, ABC):
                 "enabled": self.enabled,
             }
         )
+        if CAPTURE_NMNE:
+            state.update({"nmne": self.nmne})
         return state
 
     def reset_component_for_episode(self, episode: int):

src/primaite/simulator/network/hardware/nodes/host/host_node.py

@@ -205,7 +205,7 @@ class NIC(IPWiredNetworkInterface):
         state = super().describe_state()
 
         # Update the state with NIC-specific information
-        state.update({"wake_on_lan": self.wake_on_lan, "nmne": self.nmne})
+        state.update({"wake_on_lan": self.wake_on_lan})
 
         return state