refactor(imports): enforce guard pattern (#3382)

* refactor(imports): enforce guard pattern

* fix(tests): skip reachy2 if not installed

* Address review feedback
This commit is contained in:
Steven Palma
2026-04-14 22:54:05 +02:00
committed by GitHub
parent 60e7d67cb8
commit 9021d2d240
30 changed files with 397 additions and 290 deletions
+2 -1
View File
@@ -26,7 +26,7 @@ import torch
import torch.nn.functional as F # noqa: N812
from torch import Tensor, nn
from lerobot.utils.import_utils import _transformers_available
from lerobot.utils.import_utils import _transformers_available, require_package
# Conditional import for type checking and lazy loading
if TYPE_CHECKING or _transformers_available:
@@ -918,6 +918,7 @@ class PI05Policy(PreTrainedPolicy):
Args:
config: Policy configuration class instance.
"""
require_package("transformers", extra="pi")
super().__init__(config)
config.validate_features()
self.config = config