From 344ee29ffc069664f9e4779efd8d84db8ee29324 Mon Sep 17 00:00:00 2001 From: Kale-ab Tessera Date: Tue, 7 Nov 2023 12:24:12 +0000 Subject: [PATCH] fix: add black death wrapper to test. --- test/test_vector/test_pettingzoo_to_vec.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/test_vector/test_pettingzoo_to_vec.py b/test/test_vector/test_pettingzoo_to_vec.py index e05d09c..bd379e7 100644 --- a/test/test_vector/test_pettingzoo_to_vec.py +++ b/test/test_vector/test_pettingzoo_to_vec.py @@ -98,6 +98,7 @@ def test_terminal_obs_are_returned(): """ max_cycles = 300 env = knights_archers_zombies_v10.parallel_env(spawn_rate=50, max_cycles=300) + env = black_death_v3(env) env = pettingzoo_env_to_vec_env_v1(env) env.reset(seed=42)