Fix direct evaluator import path

script.py

@@ -1,8 +1,12 @@
 from __future__ import annotations
 
 import os
+import sys
 from pathlib import Path
 
+# Makes the local src package available when the evaluator runs script.py directly.
+sys.path.insert(0, str(Path(__file__).resolve().parent / "src"))
+
 import joblib
 import pandas as pd
 from batteryswap_public.utils import iterate_scenarios, load_dataset
@@ -17,16 +21,27 @@ def main() -> None:
     )
     splits = os.environ.get("BATTERYSWAP_SPLITS", "public,private").split(",")
     output_path = Path(os.environ.get("BATTERYSWAP_SUBMISSION_PATH", "submission.csv"))
+
     planner = joblib.load(artifact_path)
 
     plans = []
     for split in splits:
         locations, timeseries, hidden_eol_times, scenarios = load_dataset(dataset_path / split)
         daily = build_daily_features(timeseries)
+
         for scenario, locs, _, _ in iterate_scenarios(
-            locations, timeseries, hidden_eol_times, scenarios
+            locations,
+            timeseries,
+            hidden_eol_times,
+            scenarios,
         ):
-            snapshot = scenario_snapshot(daily, locs, scenario["name"], scenario["start_time"])
+            snapshot = scenario_snapshot(
+                daily,
+                locs,
+                scenario["name"],
+                scenario["start_time"],
+            )
+
             plan = planner.plan_snapshot(
                 snapshot,
                 locs,
@@ -34,15 +49,17 @@ def main() -> None:
                 scenario["settings"],
                 scenario["start_time"],
             )
+
             plan["split"] = split
             plan["scenario"] = scenario["name"]
             plans.append(plan)
 
     submission = pd.concat(plans, ignore_index=True)
     submission.to_csv(output_path, index=False)
+
     if not output_path.exists():
         raise RuntimeError(f"Submission was not created: {output_path}")
 
 
 if __name__ == "__main__":
-    main()
+    main()