Import pybullet_envs

This commit is contained in:
Shangtong Zhang
2018-05-19 22:10:32 -06:00
parent 98d7a8a482
commit 7066fb8e89
+1
View File
@@ -123,6 +123,7 @@ class Bullet(BaseTask):
class PixelBullet(BaseTask):
def __init__(self, name, seed=0, log_dir=None, frame_skip=4, history_length=4):
import pybullet_envs
self.name = name
env = gym.make(name)
env.seed(seed)