diff --git a/verse/agents/example_agent/car_agent.py b/verse/agents/example_agent/car_agent.py index 353ccc6dfa33b549915bc19720e2007882f1bab6..d70aa84fe9c570b5e1aa49efd4d9a9f41d2c18fb 100644 --- a/verse/agents/example_agent/car_agent.py +++ b/verse/agents/example_agent/car_agent.py @@ -44,7 +44,7 @@ class NPCAgent(BaseAgent): a = 0 return steering, a - def TC_simulate(self, mode: List[str], initialCondition, time_bound, time_step, lane_map:LaneMap=None)->TraceType: + def TC_simulate(self, mode: Tuple[str], initialCondition, time_bound, time_step, lane_map:LaneMap=None)->TraceType: time_bound = float(time_bound) number_points = int(np.ceil(time_bound/time_step)) diff --git a/verse/analysis/analysis_tree.py b/verse/analysis/analysis_tree.py index 1e8601d02ebaa06f48095bc272196ec673d7c619..dcfe954ff56a7d264f89a05263433aa84344cedd 100644 --- a/verse/analysis/analysis_tree.py +++ b/verse/analysis/analysis_tree.py @@ -1,6 +1,6 @@ from functools import reduce import pickle -from typing import List, Dict, Any, Optional +from typing import List, Dict, Any, Optional, Tuple import json from treelib import Tree import numpy.typing as nptyp, numpy as np, portion @@ -35,7 +35,7 @@ class AnalysisTreeNode: ): self.trace: Dict[str, TraceType] = trace self.init: Dict[str, List[float]] = init - self.mode: Dict[str, List[str]] = mode + self.mode: Dict[str, Tuple[str]] = mode self.agent: Dict = agent self.height: int = height self.child: List[AnalysisTreeNode] = child