Remove checking for specific pytorch warnings that don't seem to appe…

…ar anymore during DAgger tests.
HumanCompatibleAI · Dec 18, 2023 · b74ad0f · b74ad0f
1 parent a55ff9e
commit b74ad0f
Showing 1 changed file with 8 additions and 16 deletions.
diff --git a/tests/scripts/test_scripts.py b/tests/scripts/test_scripts.py
@@ -266,22 +266,14 @@ def test_train_preference_comparisons_reward_named_config(tmpdir, named_configs)
 
 
 def test_train_dagger_main(tmpdir):
-    with pytest.warns(None) as record:
-        run = train_imitation.train_imitation_ex.run(
-            command_name="dagger",
-            named_configs=["seals_cartpole"] + ALGO_FAST_CONFIGS["imitation"],
-            config_updates=dict(
-                logging=dict(log_root=tmpdir),
-                demonstrations=dict(path=CARTPOLE_TEST_ROLLOUT_PATH),
-            ),
-        )
-    for warning in record:
-        # PyTorch wants writeable arrays.
-        # See https://github.com/HumanCompatibleAI/imitation/issues/219
-        assert not (
-            warning.category == UserWarning
-            and "NumPy array is not writeable" in warning.message.args[0]
-        )
+    run = train_imitation.train_imitation_ex.run(
+        command_name="dagger",
+        named_configs=["seals_cartpole"] + ALGO_FAST_CONFIGS["imitation"],
+        config_updates=dict(
+            logging=dict(log_root=tmpdir),
+            demonstrations=dict(path=CARTPOLE_TEST_ROLLOUT_PATH),
+        ),
+    )
     assert run.status == "COMPLETED"
     assert isinstance(run.result, dict)