Make interactive.py more usable

Author: shariqiqbal2810

bin/interactive.py

@@ -2,6 +2,7 @@
 import os,sys
 sys.path.insert(1, os.path.join(sys.path[0], '..'))
 import argparse
+import time
 
 from multiagent.environment import MultiAgentEnv
 from multiagent.policy import InteractivePolicy
@@ -18,14 +19,17 @@
  # create world
  world = scenario.make_world()
  # create multiagent environment
- env = MultiAgentEnv(world, scenario.reset_world, scenario.reward, scenario.observation, info_callback=None, shared_viewer = False)
+ env = MultiAgentEnv(world, scenario.reset_world, scenario.reward,
+ scenario.observation, info_callback=None,
+ shared_viewer=False, discrete_action=True)
  # render call to create viewer window (necessary only for interactive policies)
  env.render()
  # create interactive policies for each agent
  policies = [InteractivePolicy(env,i) for i in range(env.n)]
  # execution loop
  obs_n = env.reset()
  while True:
+ start = time.time()
  # query for action from each agent's policy
  act_n = []
  for i, policy in enumerate(policies):
@@ -34,6 +38,9 @@
  obs_n, reward_n, done_n, _ = env.step(act_n)
  # render all agent views
  env.render()
+ end = time.time()
+ elapsed = end - start
+ time.sleep(max(1 / 30 - elapsed, 0))
  # display rewards
  #for agent in env.world.agents:
  # print(agent.name + " reward: %0.3f" % env._get_reward(agent))