Merge pull request #39 from instadeepai/lbeyers-patch-bool-conversion

Update subsample_similar.py

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]