Adapt base_ac.py and utils.py to be compatible with refactored environment

This commit is contained in:
Julian Schönberger
2024-03-27 17:04:14 +01:00
parent 1e4ec254f4
commit 086a921929
4 changed files with 66 additions and 23 deletions

15
studies/marl_adapted.py Normal file
View File

@ -0,0 +1,15 @@
from pathlib import Path
from marl_factory_grid.algorithms.marl.iac import LoopIAC
from marl_factory_grid.algorithms.utils import load_yaml_file
if __name__ == '__main__':
cfg_path = Path('../marl_factory_grid/algorithms/marl/example_config.yaml')
cfg = load_yaml_file(cfg_path)
print("Training phase")
agent = LoopIAC(cfg)
agent.train_loop()
print("Evaluation phase")
agent.eval_loop(10)