Update sim_manager.py

NREL · Jan 22, 2025 · ff8d0c0 · ff8d0c0
1 parent ff4cd5c
commit ff8d0c0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/altrios/sim_manager.py b/python/altrios/sim_manager.py
@@ -137,6 +137,7 @@ def main(
          'Arrival_Time_Actual_Hr': pl.Series([this[len(this)-1].time_hours for this in timed_paths], dtype=pl.Float64)}
     )
 
+    train_consist_plan_untrimmed = train_consist_plan.clone()
     train_consist_plan = (train_consist_plan
         .join(train_times,on=["Train_ID","Origin_ID","Destination_ID"],how="left")
     )
@@ -165,4 +166,5 @@ def main(
         nodal_energy_prices,
         train_sims,
         timed_paths,
+        train_consist_plan_untrimmed
     )