Time-Space Diagram offset axes

examples/exp_configs/non_rl/i210_subnetwork.py

@@ -108,33 +108,33 @@
         veh_type="human",
         edge=highway_start_edge,
         vehs_per_hour=INFLOW_RATE * (1 - PENETRATION_RATE),
-        departLane=lane,
-        departSpeed=INFLOW_SPEED)
+        depart_lane=lane,
+        depart_speed=INFLOW_SPEED)
 
     if PENETRATION_RATE > 0.0:
         inflow.add(
             veh_type="av",
             edge=highway_start_edge,
             vehs_per_hour=INFLOW_RATE * PENETRATION_RATE,
-            departLane=lane,
-            departSpeed=INFLOW_SPEED)
+            depart_lane=lane,
+            depart_speed=INFLOW_SPEED)
 
 # on ramp
 if ON_RAMP:
     inflow.add(
         veh_type="human",
         edge="27414345",
         vehs_per_hour=int(ON_RAMP_INFLOW_RATE * (1 - PENETRATION_RATE)),
-        departSpeed=10,
+        depart_speed=10,
     )
 
     if PENETRATION_RATE > 0.0:
         inflow.add(
             veh_type="av",
             edge="27414345",
             vehs_per_hour=int(ON_RAMP_INFLOW_RATE * PENETRATION_RATE),
-            departLane="random",
-            departSpeed=10)
+            depart_lane="random",
+            depart_speed=10)
 
 # =========================================================================== #
 # Generate the flow_params dict with all relevant simulation information.     #

flow/core/experiment.py

@@ -248,10 +248,13 @@ def rl_actions(*_):
             write_dict_to_csv(metadata_table_path, metadata, True)
             tsd_main(
                 trajectory_table_path,
-                {'network': self.env.network.__class__},
+                {
+                    'network': self.env.network.__class__,
+                    'env': self.env.env_params,
+                    'sim': self.env.sim_params
+                },
                 min_speed=0,
-                max_speed=10,
-                start=self.env.env_params.warmup_steps
+                max_speed=10
             )
             upload_to_s3(
                 'circles.data.pipeline',