From 43bef1d91cc5da7dc0180a97bf74e8a2d7a7d219 Mon Sep 17 00:00:00 2001
From: Pepijn <pepijn@huggingface.co>
Date: Mon, 13 Oct 2025 17:59:59 +0200
Subject: [PATCH] fix test

---
 tests/training/test_multi_gpu.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/training/test_multi_gpu.py b/tests/training/test_multi_gpu.py
index f0668a8ef..5aedde374 100644
--- a/tests/training/test_multi_gpu.py
+++ b/tests/training/test_multi_gpu.py
@@ -119,7 +119,7 @@ class TestMultiGPUTraining:
             config_args = [
                 "--dataset.repo_id=lerobot/pusht",
                 "--dataset.episodes=[0]",
-                "--policy=act",
+                "--policy.type=act",
                 "--policy.device=cuda",
                 f"--output_dir={output_dir}",
                 "--batch_size=4",
@@ -157,7 +157,7 @@ class TestMultiGPUTraining:
             config_args = [
                 "--dataset.repo_id=lerobot/pusht",
                 "--dataset.episodes=[0]",
-                "--policy=act",
+                "--policy.type=act",
                 "--policy.device=cuda",
                 f"--output_dir={output_dir}",
                 "--batch_size=4",