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)})