mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-16 09:09:48 +00:00
pre-commit
This commit is contained in:
@@ -22,6 +22,7 @@ import pytest
|
||||
|
||||
try:
|
||||
import peft # noqa: F401
|
||||
|
||||
WALLX_AVAILABLE = True
|
||||
except ImportError:
|
||||
WALLX_AVAILABLE = False
|
||||
@@ -34,6 +35,7 @@ pytestmark = pytest.mark.skipif(
|
||||
|
||||
if WALLX_AVAILABLE:
|
||||
import torch
|
||||
|
||||
from lerobot.policies.factory import make_policy_config
|
||||
from lerobot.policies.wall_x import (
|
||||
WallXConfig,
|
||||
|
||||
Reference in New Issue
Block a user