From 9ea8bd029c6633caedb84a9d1af2f4dec1ee68ad Mon Sep 17 00:00:00 2001
From: Pepijn <pepijn@huggingface.co>
Date: Wed, 10 Sep 2025 19:50:49 +0200
Subject: [PATCH] change device in test

---
 test_pi0_openpi.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test_pi0_openpi.py b/test_pi0_openpi.py
index d7e5ffc0b..ad8419f7a 100644
--- a/test_pi0_openpi.py
+++ b/test_pi0_openpi.py
@@ -13,7 +13,7 @@ def test_policy_instantiation():
     print("Testing PI0OpenPI policy instantiation...")
 
     # Create config
-    config = PI0OpenPIConfig(action_dim=7, state_dim=14, device="cpu", dtype="float32")
+    config = PI0OpenPIConfig(action_dim=7, state_dim=14, dtype="float32")
 
     # Create dummy dataset stats
     dataset_stats = {