my update

This commit is contained in:
romue
2021-11-11 10:59:13 +01:00
parent 6287380f60
commit ea4582a59e
10 changed files with 88 additions and 308 deletions

View File

@ -1,9 +1,11 @@
import numpy as np
from environments.factory import rooms
import random
from gym.wrappers import FrameStack
env = rooms(n_agents=2)
env = FrameStack(env, num_stack=3)
state, *_ = env.reset()
class SatMad(object):
def __init__(self):
pass
if __name__ == '__main__':
pass
for i in range(1000):
state, *_ = env.step([random.randint(0, 9), random.randint(0, 9)])
env.render()