From c9bc8fbf3d512f2ba3fecec1ff008ad2e8a57bb3 Mon Sep 17 00:00:00 2001 From: Marek Wolan Date: Mon, 9 Oct 2023 18:33:30 +0100 Subject: [PATCH] Fix file observation test --- tests/integration_tests/game_layer/test_observations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integration_tests/game_layer/test_observations.py b/tests/integration_tests/game_layer/test_observations.py index 7f20a938..c1f20d78 100644 --- a/tests/integration_tests/game_layer/test_observations.py +++ b/tests/integration_tests/game_layer/test_observations.py @@ -16,5 +16,5 @@ def test_file_observation(): dog_file_obs = FileObservation( where=["network", "nodes", pc.uuid, "file_system", "folders", "root", "files", "dog.png"] ) - assert dog_file_obs(state) == {"health_status": 1} + assert dog_file_obs.observe(state) == {"health_status": 1} assert dog_file_obs.space == spaces.Dict({"health_status": spaces.Discrete(6)})