From 9197e89d42dd40f449ca4fe5a6159a6c4b008661 Mon Sep 17 00:00:00 2001
From: sakhtar <sakhtar@nrel.gov>
Date: Mon, 23 Sep 2024 16:35:45 -0600
Subject: [PATCH] Corrected return type for drag_coeff_function field

---
 python/altrios/train_planner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/altrios/train_planner.py b/python/altrios/train_planner.py
index 0ac2bcde..b42b2f83 100644
--- a/python/altrios/train_planner.py
+++ b/python/altrios/train_planner.py
@@ -50,7 +50,7 @@ class TrainPlannerConfig:
     refuelers_per_incoming_corridor: int = 4
     require_diesel: bool = False
     manifest_empty_return_ratio: float = 0.6
-    drag_coeff_function: List = field(default_factory = lambda: None)
+    drag_coeff_function: Callable = None
     hp_required_per_ton: Dict = field(default_factory = lambda: {
         "Default": {
         "Unit": 2.0,