admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-22 03:59:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

pre-commit

Browse Source
This commit is contained in:
Pepijn
2025-12-17 16:04:03 +01:00
committed by Michel Aractingi
parent 1ba51a6d02
commit 3434a5d5df
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/wall_x/test_wallx.py
+2
View File
@@ -22,6 +22,7 @@ import pytest
try: try:
import peft # noqa: F401 import peft # noqa: F401
WALLX_AVAILABLE = True WALLX_AVAILABLE = True
except ImportError: except ImportError:
WALLX_AVAILABLE = False WALLX_AVAILABLE = False
@@ -34,6 +35,7 @@ pytestmark = pytest.mark.skipif(
if WALLX_AVAILABLE: if WALLX_AVAILABLE:
import torch import torch
from lerobot.policies.factory import make_policy_config from lerobot.policies.factory import make_policy_config
from lerobot.policies.wall_x import ( from lerobot.policies.wall_x import (
WallXConfig, WallXConfig,