my update
This commit is contained in:
@ -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()
|
Reference in New Issue
Block a user