oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1443 - reverted changes made to observation space and added config files for testing

Browse Source
This commit is contained in:
SunilSamra
2023-05-31 14:11:15 +01:00
parent a8cc50a495
commit 81e9ddca9b
4 changed files with 31 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_single_action_space.py
View File

@@ -5,8 +5,8 @@ from tests.conftest import _get_primaite_env_from_config
def test_single_action_space():
"""Test to ensure the blue agent is using the ACL action space and is carrying out both kinds of operations."""
env = _get_primaite_env_from_config(
main_config_path=TEST_CONFIG_ROOT / "one_node_states_on_off_main_config.yaml",
main_config_path=TEST_CONFIG_ROOT / "single_action_space_main_config.yaml",
lay_down_config_path=TEST_CONFIG_ROOT
/ "one_node_states_on_off_lay_down_config.yaml",
/ "single_action_space_lay_down_config.yaml",
)
print("Average Reward:", env.average_reward)