admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-17 09:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix testing

Browse Source
This commit is contained in:
Jade Choghari
2025-11-25 21:11:17 +01:00
parent 4e9acd4afe
commit 15dc2fd867
2 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/xvla/test_xvla_original_vs_lerobot.py
+12 -3
View File
@@ -26,9 +26,18 @@ import numpy as np
import pytest
import torch
from lerobot.policies.xvla.configuration_xvla import XVLAConfig
from lerobot.policies.xvla.modeling_xvla import XVLAPolicy
from lerobot.policies.xvla.processor_xvla import make_xvla_pre_post_processors
# Conditional import for type checking and lazy loading
from lerobot.utils.import_utils import _timm_available, _transformers_available
if _timm_available and _transformers_available:
from lerobot.policies.xvla.configuration_xvla import XVLAConfig
from lerobot.policies.xvla.modeling_xvla import XVLAPolicy
from lerobot.policies.xvla.processor_xvla import make_xvla_pre_post_processors
else:
XVLAConfig = None
XVLAPolicy = None
make_xvla_pre_post_processors = None
from lerobot.processor import PolicyAction, PolicyProcessorPipeline # noqa: E402
from lerobot.utils.constants import OBS_IMAGES, OBS_STATE # noqa: E402
from tests.utils import require_package # noqa: E402