Fix policy testing for tv5 (#3032)

* fix ci logger

* other fix

* fix mypy

* change logits to torch2.10

* skip wallx|

* remove logging

---------

Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>

tests/policies/pi0_fast/test_pi0_fast_original_vs_lerobot.py

@@ -60,13 +60,13 @@ MODEL_PATH_LEROBOT = "jadechoghari/pi0fast-base"
 EXPECTED_ACTION_TOKENS_SHAPE = (1, 2)
 
 # Expected first 5 action tokens (for reproducibility check)
-EXPECTED_ACTION_TOKENS_FIRST_5 = torch.tensor([255657, 255425])
+EXPECTED_ACTION_TOKENS_FIRST_5 = torch.tensor([255020, 255589])
 
 # Expected actions after detokenization
 EXPECTED_ACTIONS_SHAPE = (1, 2, 32)  # (batch_size, n_action_steps, action_dim)
 EXPECTED_ACTIONS_MEAN = 0.046403881162405014
 EXPECTED_ACTIONS_STD = 0.2607129216194153
-EXPECTED_ACTIONS_FIRST_5 = torch.tensor([-0.0707, 1.4849, 0.0000, 0.0000, 0.0000])
+EXPECTED_ACTIONS_FIRST_5 = torch.tensor([0.0000, 0.3536, 0.0707, 0.0000, 0.0000])
 
 
 def set_seed_all(seed: int):

tests/policies/wall_x/test_wallx.py

@@ -16,9 +16,15 @@
 
 """Test script to verify Wall-X policy integration with LeRobot, only meant to be run locally!"""
 
+import os
+
 import pytest
 import torch
 
+pytestmark = pytest.mark.skipif(
+    os.environ.get("CI") == "true" or os.environ.get("GITHUB_ACTIONS") == "true",
+    reason="This test exceeds available memory in CI environments.",
+)
 # Skip if required dependencies are not available
 pytest.importorskip("peft")
 pytest.importorskip("transformers")