vishanth11 commited on
Commit
eb1ca88
·
1 Parent(s): 15cfba5

fix: Add traci/sumolib to requirements and safe fallback imports

Browse files
requirements.txt CHANGED
@@ -28,7 +28,9 @@ shap>=0.45.0
28
  matplotlib>=3.8.0
29
  seaborn>=0.13.0
30
 
31
- # Utilities
32
  pyyaml>=6.0.1
33
  requests>=2.31.0
34
  tqdm>=4.66.0
 
 
 
28
  matplotlib>=3.8.0
29
  seaborn>=0.13.0
30
 
31
+ # Utilities & Simulation
32
  pyyaml>=6.0.1
33
  requests>=2.31.0
34
  tqdm>=4.66.0
35
+ traci>=1.18.0
36
+ sumolib>=1.18.0
src/integration/sumo_conflict_adapter.py CHANGED
@@ -1,4 +1,7 @@
1
- import traci
 
 
 
2
 
3
  from integration.conflict_graph import ConflictGraph
4
  from integration.sumo_topology_adapter import SUMOTopologyAdapter
 
1
+ try:
2
+ import traci
3
+ except ImportError:
4
+ traci = None
5
 
6
  from integration.conflict_graph import ConflictGraph
7
  from integration.sumo_topology_adapter import SUMOTopologyAdapter
src/integration/sumo_decision_controller.py CHANGED
@@ -21,8 +21,10 @@ if SRC_ROOT not in sys.path:
21
  sys.path.insert(0, SRC_ROOT)
22
  if PROJECT_ROOT not in sys.path:
23
  sys.path.insert(0, PROJECT_ROOT)
24
-
25
- import traci
 
 
26
 
27
  from integration.controller import TrafficController
28
  from integration.decision_engine import DecisionEngine
 
21
  sys.path.insert(0, SRC_ROOT)
22
  if PROJECT_ROOT not in sys.path:
23
  sys.path.insert(0, PROJECT_ROOT)
24
+ try:
25
+ import traci
26
+ except ImportError:
27
+ traci = None
28
 
29
  from integration.controller import TrafficController
30
  from integration.decision_engine import DecisionEngine
src/integration/sumo_demand.py CHANGED
@@ -1,4 +1,7 @@
1
- import traci
 
 
 
2
 
3
 
4
  class SUMODemandCalculator:
 
1
+ try:
2
+ import traci
3
+ except ImportError:
4
+ traci = None
5
 
6
 
7
  class SUMODemandCalculator:
src/integration/sumo_phase_adapter.py CHANGED
@@ -1,4 +1,7 @@
1
- import traci
 
 
 
2
 
3
 
4
  class SUMOPhaseAdapter:
 
1
+ try:
2
+ import traci
3
+ except ImportError:
4
+ traci = None
5
 
6
 
7
  class SUMOPhaseAdapter:
src/integration/sumo_prediction_controller.py CHANGED
@@ -1,7 +1,10 @@
1
  import os
2
  import sys
3
 
4
- import traci
 
 
 
5
  import numpy as np
6
 
7
  from integration.prediction_adapter import TrafficPredictionAdapter
 
1
  import os
2
  import sys
3
 
4
+ try:
5
+ import traci
6
+ except ImportError:
7
+ traci = None
8
  import numpy as np
9
 
10
  from integration.prediction_adapter import TrafficPredictionAdapter
src/integration/sumo_topology_adapter.py CHANGED
@@ -1,4 +1,7 @@
1
- import traci
 
 
 
2
 
3
  from integration.topology import JunctionTopology
4
 
 
1
+ try:
2
+ import traci
3
+ except ImportError:
4
+ traci = None
5
 
6
  from integration.topology import JunctionTopology
7