Fix tests and type test.

huawei-noah · Dec 21, 2023 · 2b088b2 · 2b088b2
1 parent d4ffea6
commit 2b088b2
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 1,158 deletions.
diff --git a/examples/direct/bubble_example.py b/examples/direct/bubble_example.py
@@ -177,7 +177,7 @@ class ObservationState:
         def __init__(self) -> None:
             self.last_observations: Dict[str, Observation] = {}
 
-        def observation_callback(self, obs: Observation):
+        def observation_callback(self, obs: Dict[str, Observation]):
             self.last_observations = obs
 
     obs_state = ObservationState()