From e9e7eb827a8cde55baa7770b298f16db1f7c8de9 Mon Sep 17 00:00:00 2001
From: Pepijn <pepijn@huggingface.co>
Date: Wed, 10 Sep 2025 21:36:58 +0200
Subject: [PATCH] also shorten action_steps

---
 test_pi05_openpi.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/test_pi05_openpi.py b/test_pi05_openpi.py
index 60f51c92f..4393d088c 100644
--- a/test_pi05_openpi.py
+++ b/test_pi05_openpi.py
@@ -90,6 +90,7 @@ def test_pi05_forward_pass():
         dtype="float32",
         pi05=True,
         action_horizon=16,  # Shorter horizon for testing
+        n_action_steps=16,  # Shorter action steps for testing
     )
 
     # Create dummy dataset stats