From 2dcbef99c1f3f48ce1dfa12dcaf57388f9bce7e9 Mon Sep 17 00:00:00 2001 From: Antonin Raffin Date: Sat, 13 Aug 2022 15:22:13 +0200 Subject: [PATCH] Update test --- tests/test_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_utils.py b/tests/test_utils.py index 67f2ad1..0d1c144 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -384,9 +384,10 @@ def test_get_system_info(): assert info["Stable-Baselines3"] == str(sb3.__version__) assert "Python" in info_str assert "PyTorch" in info_str - assert "GPU Enabled" in info_str + assert "Accelerator" in info_str assert "Numpy" in info_str assert "Gym" in info_str + assert "Cloudpickle" in info_str def test_is_vectorized_observation():