saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-02 03:55:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update test

Browse source
This commit is contained in:
Antonin Raffin 2022-08-13 15:22:13 +02:00
parent 9ac6225cdb
commit 2dcbef99c1
No known key found for this signature in database
GPG key ID: B8B48F65CAD6232C
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_utils.py
View file

@ -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():