From 252bca9354a005d1741dbef29b4b9e08c21ab67f Mon Sep 17 00:00:00 2001 From: Pepijn Date: Tue, 14 Oct 2025 10:06:32 +0200 Subject: [PATCH] dont push to hub in multi gpu tests --- tests/training/test_multi_gpu.py | 2 ++ 1 file changed, 2 insertions(+) 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",