#2533: clean up temp items in tests + fixing the CLI tests

This commit is contained in:
Czar Echavez
2024-05-01 14:33:33 +01:00
parent a2fb04e6f6
commit b6b5ce91c2
5 changed files with 44 additions and 53 deletions

View File

@@ -38,3 +38,5 @@ def test_rllib_single_agent_compatibility():
save_file = Path(tempfile.gettempdir()) / "ray/"
algo.save(save_file)
assert save_file.exists()
save_file.unlink() # clean up

View File

@@ -25,3 +25,4 @@ def test_sb3_compatibility():
model.save(save_path)
assert (save_path).exists()
save_path.unlink() # clean up