diff --git a/tests/training/test_multi_gpu.py b/tests/training/test_multi_gpu.py index 5aedde374..73c0d39dd 100644 --- a/tests/training/test_multi_gpu.py +++ b/tests/training/test_multi_gpu.py @@ -121,6 +121,7 @@ class TestMultiGPUTraining: "--dataset.episodes=[0]", "--policy.type=act", "--policy.device=cuda", + "--policy.push_to_hub=false", f"--output_dir={output_dir}", "--batch_size=4", "--steps=10", @@ -159,6 +160,7 @@ class TestMultiGPUTraining: "--dataset.episodes=[0]", "--policy.type=act", "--policy.device=cuda", + "--policy.push_to_hub=false", f"--output_dir={output_dir}", "--batch_size=4", "--steps=20",