Updated agents to have type hints

f1tenth · Dec 12, 2023 · 0e61411 · 0e61411
1 parent 170bd71
commit 0e61411
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gym/f110_gym/envs/base_classes.py b/gym/f110_gym/envs/base_classes.py
@@ -26,6 +26,7 @@
 Replacement of the old RaceCar, Simulator classes in C++
 Author: Hongrui Zheng
 """
+from typing import List
 import numpy as np
 from f110_gym.envs.dynamic_models import DynamicModel
 from f110_gym.envs.action import CarAction
@@ -406,7 +407,7 @@ def __init__(
         self.ego_idx = ego_idx
         self.params = params
         self.agent_poses = np.empty((self.num_agents, 3))
-        self.agents = []
+        self.agents: List[RaceCar] = []
         self.collisions = np.zeros((self.num_agents,))
         self.collision_idx = -1 * np.ones((self.num_agents,))
         self.model = model