98fc1e4c71
- renamed _config_values_main to training_config.py and renamed the ConfigValuesMain class to TrainingConfig. Moved training_config.py to src/primaite/config/training_config.py - Renamed all training config yaml file keys to make creating an instance of TrainingConfig easier. Moved action_type and num_steps over to the training config. - Decoupled the training config and lay down config. - Refactored main.py so that it can be ran from CLI and can take a training config path and a lay down config path. - refactored all outputs so that they save to the session dir. - Added some necessary setup scripts that handle creating app dirs, fronting example config files to the user, fronting demo notebooks to the user, performing clean-up in between installations etc. - Added functions that attempt to retrieve the file path of users example config files that have been fronted by the primaite setup. - Added logging config and a getLogger function in the top-level init. - Refactored all logs entries logged to use a logger using the primaite logging config. - Added basic typer CLI for doing things like setup, viewing logs, viewing primaite version, running a basic session. - Updated test to use new features and config structures. - Began updating docs. More to do here.
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
"""Test env creation and behaviour with different observation spaces."""
|
|
|
|
from tests import TEST_CONFIG_ROOT
|
|
from tests.conftest import _get_primaite_env_from_config
|
|
|
|
|
|
def test_creating_env_with_box_obs():
|
|
"""Try creating env with box observation space."""
|
|
env = _get_primaite_env_from_config(
|
|
training_config_path=TEST_CONFIG_ROOT
|
|
/ "one_node_states_on_off_main_config.yaml",
|
|
lay_down_config_path=TEST_CONFIG_ROOT / "box_obs_space_laydown_config.yaml",
|
|
)
|
|
env.update_environent_obs()
|
|
|
|
# we have three nodes and two links, with one service
|
|
# therefore the box observation space will have:
|
|
# * 5 columns (four fixed and one for the service)
|
|
# * 5 rows (3 nodes + 2 links)
|
|
assert env.env_obs.shape == (5, 5)
|
|
|
|
|
|
def test_creating_env_with_multidiscrete_obs():
|
|
"""Try creating env with MultiDiscrete observation space."""
|
|
env = _get_primaite_env_from_config(
|
|
training_config_path=TEST_CONFIG_ROOT
|
|
/ "one_node_states_on_off_main_config.yaml",
|
|
lay_down_config_path=TEST_CONFIG_ROOT
|
|
/ "multidiscrete_obs_space_laydown_config.yaml",
|
|
)
|
|
env.update_environent_obs()
|
|
|
|
# we have three nodes and two links, with one service
|
|
# the nodes have hardware, OS, FS, and service, the links just have bandwidth,
|
|
# therefore we need 3*4 + 2 observations
|
|
assert env.env_obs.shape == (3 * 4 + 2,)
|