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>
This commit is contained in:
Jade Choghari
2026-02-27 16:45:40 +03:00
committed by GitHub
parent 57d899d84c
commit 3cbefa445b
2 changed files with 8 additions and 2 deletions
+6
View File
@@ -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")