maintainer test grabs temp state

This commit is contained in:
Chanumask
2023-11-23 12:52:27 +01:00
parent 3a7b727ec6
commit 4c85cac50f
9 changed files with 82 additions and 28 deletions

View File

@ -87,4 +87,5 @@ Rules:
max_steps: 500
Tests:
MaintainerTest: {}
MaintainerTest: {}
# DirtAgentTest: {}

View File

@ -3,6 +3,7 @@ from random import randint
from tqdm import trange
from marl_factory_grid.algorithms.static.TSP_dirt_agent import TSPDirtAgent
from marl_factory_grid.environment.factory import Factory
if __name__ == '__main__':
@ -21,6 +22,7 @@ if __name__ == '__main__':
if render:
factory.render()
action_spaces = factory.action_space
# agents = [TSPDirtAgent(factory, 0)]
while not done:
a = [randint(0, x.n - 1) for x in action_spaces]
obs_type, _, _, done, info = factory.step(a)