update mpi usage

This commit is contained in:
Erwin Coumans
2019-02-01 19:20:08 -08:00
parent 176edeff1e
commit 2eace2f715
2 changed files with 8 additions and 5 deletions

View File

@@ -40,8 +40,8 @@ def main():
# Command line arguments
args = sys.argv[1:]
world = build_world(args, enable_draw=False)
enable_draw = False
world = build_world(args, enable_draw)
run()
shutdown()

View File

@@ -15,6 +15,7 @@ from pybullet_envs.deep_mimic.env.pybullet_deep_mimic_env import PyBulletDeepMim
import sys
import random
update_timestep = 1./600.
def update_world(world, time_elapsed):
timeStep = 1./600.
@@ -43,8 +44,9 @@ args = sys.argv[1:]
def build_world(args, enable_draw, playback_speed=1):
def build_world(args, enable_draw):
arg_parser = build_arg_parser(args)
print("enable_draw=",enable_draw)
env = PyBulletDeepMimicEnv(args, enable_draw)
world = RLWorld(env, arg_parser)
#world.env.set_playback_speed(playback_speed)
@@ -72,8 +74,9 @@ def build_world(args, enable_draw, playback_speed=1):
world.reset()
return world
world = build_world(args, True)
while (world.env._pybullet_client.isConnected()):
if __name__ == '__main__':
world = build_world(args, True)
while (world.env._pybullet_client.isConnected()):
timeStep = 1./600.
update_world(world, timeStep)