admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-09 11:01:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

adding lazy imports

Browse Source
This commit is contained in:
Maxime Ellerbach
2026-06-18 14:17:09 +00:00
parent 7049887017
commit 2f12f58e33
4 changed files with 41 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/policies/fastwam/test_fastwam_policy.py
+3
View File
@@ -21,6 +21,9 @@ import torch
from safetensors import safe_open
from torch import nn
pytest.importorskip("transformers", reason="fastwam requires the `fastwam` extra (transformers)")
pytest.importorskip("diffusers", reason="fastwam requires the `fastwam` extra (diffusers)")
from lerobot.configs import FeatureType, PolicyFeature, PreTrainedConfig
from lerobot.policies import FastWAMConfig, get_policy_class, make_policy_config, make_pre_post_processors
from lerobot.policies.fastwam.modeling_fastwam import FastWAMPolicy