admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 12:09:42 +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
src/lerobot/utils/import_utils.py
+1
View File
@@ -62,6 +62,7 @@ def is_package_available(pkg_name: str, return_version: bool = False) -> tuple[b
_transformers_available = is_package_available("transformers")
_timm_available = is_package_available("timm")
_peft_available = is_package_available("peft")
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