Update subsample_similar.py

instadeepai · Aug 23, 2024 · 645078c · 645078c
1 parent 9fbc536
commit 645078c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/og_marl/vault_utils/subsample_similar.py b/og_marl/vault_utils/subsample_similar.py
@@ -35,7 +35,7 @@ def get_episode_returns_and_term_idxes(offline_data: Dict[str, Array]) -> Tuple[
     rewards = offline_data["rewards"][0, :, 0]
     terminal_flag = offline_data["terminals"][0, :, ...].all(axis=-1)
 
-    assert terminal_flag[-1] is True
+    assert bool(terminal_flag[-1]) is True
 
     def scan_cumsum(
         return_so_far: float, prev_term_reward: Tuple[bool, float]