fix previous commit

2024-03-27 22:11:37 +00:00
parent cae9f64b93
commit 0d0b5bc7d9
2 changed files with 2 additions and 2 deletions
--- a/tests/integration_tests/game_layer/observations/test_node_observations.py
+++ b/tests/integration_tests/game_layer/observations/test_node_observations.py
@@ -4,7 +4,7 @@ from uuid import uuid4
 import pytest
 from gymnasium import spaces

-from primaite.game.agent.observations.host import NodeObservation
+from primaite.game.agent.observations.node_observations import NodeObservation
 from primaite.simulator.network.hardware.nodes.host.computer import Computer
 from primaite.simulator.sim_container import Simulation