eb7fb37806
* update atari.py * fix setup.py pass the pytest * fix setup.py pass the pytest * add args "render" * change the tensorboard writter * change the tensorboard writter * change device, render, tensorboard log location * change device, render, tensorboard log location * remove some wrong local files * fix some tab mistakes and the envs name in continuous/test_xx.py * add examples and point robot maze environment * fix some bugs during testing examples * add dqn network and fix some args * change back the tensorboard writter's frequency to ensure ppo and a2c can write things normally * add a warning to collector * rm some unrelated files * reformat * fix a bug in test_dqn due to the model wrong selection * change atari frame skip and observation to improve performance * readd some files * change import * modified readme * rm tensorboard log * update atari and mujoco which are ignored * rm the wrong lines * readd the import of PointMaze * fix a typo in test/discrete/net.py * add a class declaration to pass the flake8 * fix flake8 errors
17 lines
405 B
Python
17 lines
405 B
Python
from tianshou.env.utils import CloudpickleWrapper
|
|
from tianshou.env.common import EnvWrapper, FrameStack
|
|
from tianshou.env.vecenv import BaseVectorEnv, VectorEnv, \
|
|
SubprocVectorEnv, RayVectorEnv
|
|
from tianshou.env import mujoco
|
|
|
|
__all__ = [
|
|
'mujoco',
|
|
'EnvWrapper',
|
|
'FrameStack',
|
|
'BaseVectorEnv',
|
|
'VectorEnv',
|
|
'SubprocVectorEnv',
|
|
'RayVectorEnv',
|
|
'CloudpickleWrapper',
|
|
]
|