Merge pull request berkeleyflow#47 from berkeleyflow/port_fix3

moved port method
kjang96 · Jun 9, 2018 · 89945a6 · 89945a6
2 parents 7928af1 + 37e3081
commit 89945a6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/flow/envs/base_env.py b/flow/envs/base_env.py
@@ -71,6 +71,9 @@ def __init__(self, env_params, sumo_params, scenario):
         self.env_params = env_params
         self.scenario = scenario
         self.sumo_params = sumo_params
+        time_stamp = ''.join(str(time.time()).split('.'))
+        time.sleep(8.0 * int(time_stamp[-6:]) / 1e6)
+        self.sumo_params.port = sumolib.miscutils.getFreeSocketPort()
         self.vehicles = scenario.vehicles
         self.traffic_lights = scenario.traffic_lights
         # time_counter: number of steps taken since the start of a rollout

diff --git a/flow/utils/rllib.py b/flow/utils/rllib.py
@@ -95,10 +95,6 @@ def create_env(*_):
         )
 
         sumo_params = deepcopy(params['sumo'])
-        if sumo_params.port is None:
-            time_stamp = ''.join(str(time.time()).split('.'))
-            time.sleep(2.0 * int(time_stamp[-6:]) / 1e6)
-            sumo_params.port = sumolib.miscutils.getFreeSocketPort()
 
         register(
             id=env_name,