From 0d0b5bc7d9fc64549f19017ded0569bfe9ba45ce Mon Sep 17 00:00:00 2001 From: Marek Wolan Date: Wed, 27 Mar 2024 22:11:37 +0000 Subject: [PATCH] fix previous commit --- src/primaite/game/agent/observations/agent_observations.py | 2 +- .../game_layer/observations/test_node_observations.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/primaite/game/agent/observations/agent_observations.py b/src/primaite/game/agent/observations/agent_observations.py index 2148697b..10370660 100644 --- a/src/primaite/game/agent/observations/agent_observations.py +++ b/src/primaite/game/agent/observations/agent_observations.py @@ -2,7 +2,7 @@ from typing import Dict, List, Optional, Tuple, TYPE_CHECKING from gymnasium import spaces -from primaite.game.agent.observations.host import NodeObservation +from primaite.game.agent.observations.node_observations import NodeObservation from primaite.game.agent.observations.observations import ( AbstractObservation, AclObservation, diff --git a/tests/integration_tests/game_layer/observations/test_node_observations.py b/tests/integration_tests/game_layer/observations/test_node_observations.py index a5195e1e..dce05b6a 100644 --- 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