From 404b8f8a7502710e0e25e371b089349a04412ede Mon Sep 17 00:00:00 2001
From: Simon Alibert <alibert.sim@gmail.com>
Date: Wed, 27 Mar 2024 12:43:30 +0100
Subject: [PATCH] Fix end-to-end ACT train on Aloha

---
 .github/workflows/test.yml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index c26234ab..7d93d43d 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -140,6 +140,8 @@ jobs:
             device=cpu \
             save_model=true \
             save_freq=2 \
+            horizon=20 \
+            policy.batch_size=2 \
             hydra.run.dir=tests/outputs/act/
 
       - name: Test eval ACT on ALOHA end-to-end