From 343579368be5bf6578a5588b39ec4232e16de162 Mon Sep 17 00:00:00 2001
From: Cadene <re.cadene@gmail.com>
Date: Wed, 24 Apr 2024 15:56:22 +0000
Subject: [PATCH] nit

---
 tests/test_policies.py | 7 -------
 1 file changed, 7 deletions(-)

diff --git a/tests/test_policies.py b/tests/test_policies.py
index d0d53e7b..0bc41ec7 100644
--- a/tests/test_policies.py
+++ b/tests/test_policies.py
@@ -194,10 +194,3 @@ def test_normalize(insert_temporal_dim):
     new_unnormalize = Unnormalize(output_shapes, unnormalize_output_modes, stats=None)
     new_unnormalize.load_state_dict(unnormalize.state_dict())
     unnormalize(output_batch)
-
-
-if __name__ == "__main__":
-    test_policy(
-        *("aloha", "act", ["env.task=AlohaTransferCube-v0", "dataset_id=aloha_sim_transfer_cube_scripted"])
-    )
-    # test_policy(insert_temporal_dim=True)