Skip to content

Commit 4f322e5

Browse files
committed
update numpy typing to be compatible with v2.0
1 parent 25d2e4b commit 4f322e5

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

simulation-system/libs/csle-common/tests/test_elk_controller.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -504,7 +504,7 @@ def test_get_elk_managers_infos(
504504
emulation_env_config.elk_config = elk_config
505505
emulation_env_config.elk_config.container.docker_gw_bridge_ip = "172.17.0.1"
506506
emulation_env_config.elk_config.elk_manager_port = 5601
507-
emulation_env_config.execution_id = "test_execution_id"
507+
emulation_env_config.execution_id = 1
508508
emulation_env_config.name = "test_emulation"
509509
logger = MagicMock(spec=logging.Logger)
510510
active_ips = ["172.17.0.1"]
@@ -520,5 +520,5 @@ def test_get_elk_managers_infos(
520520
)
521521
assert result.ips == ["172.17.0.1"]
522522
assert result.ports == [5601]
523-
assert result.execution_id == "test_execution_id"
523+
assert result.execution_id == 1
524524
assert result.emulation_name == "test_emulation"

simulation-system/libs/gym-csle-apt-game/tests/test_apt_game_env.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def test_reset(self) -> None:
133133
self.env.trace.attacker_observations = ["obs1", "obs2"]
134134
self.env.trace.defender_observations = ["obs1", "obs2"]
135135
initial_trace_count = len(self.env.traces)
136-
AptGameState.reset = lambda self: None
136+
AptGameState.reset = lambda self: None # type: ignore
137137
initial_obs, info = self.env.reset(seed=10, soft=False, options=None)
138138
assert len(self.env.traces) == initial_trace_count + 1
139139
assert isinstance(self.env.trace, SimulationTrace)

simulation-system/libs/gym-csle-apt-game/tests/test_apt_game_util.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ def test_expected_cost(self) -> None:
223223
224224
:return: None
225225
"""
226-
C = [[10, 20, 30], [15, 25, 35]]
226+
C = [[10.0, 20.0, 30.0], [15.0, 25.0, 35.0]]
227227
b = [0.2, 0.5, 0.3]
228228
S = [0, 1, 2]
229229
a1 = 0

0 commit comments

Comments
 (0)