mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-24 18:26:11 +00:00
skip molmoact2 tests without optional deps
This commit is contained in:
@@ -14,6 +14,10 @@
|
|||||||
# See the License for the specific language governing permissions and
|
# See the License for the specific language governing permissions and
|
||||||
# limitations under the License.
|
# limitations under the License.
|
||||||
|
|
||||||
|
"""Unit tests for MolmoAct2's LeRobot policy interface."""
|
||||||
|
|
||||||
|
# ruff: noqa: E402
|
||||||
|
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
import json
|
import json
|
||||||
@@ -25,6 +29,9 @@ import pytest
|
|||||||
import torch
|
import torch
|
||||||
import torch.nn.functional as F # noqa: N812
|
import torch.nn.functional as F # noqa: N812
|
||||||
|
|
||||||
|
pytest.importorskip("transformers")
|
||||||
|
pytest.importorskip("scipy")
|
||||||
|
|
||||||
from lerobot.configs import FeatureType, NormalizationMode, PolicyFeature
|
from lerobot.configs import FeatureType, NormalizationMode, PolicyFeature
|
||||||
from lerobot.policies import get_policy_class, make_policy_config
|
from lerobot.policies import get_policy_class, make_policy_config
|
||||||
from lerobot.policies.molmoact2 import (
|
from lerobot.policies.molmoact2 import (
|
||||||
|
|||||||
Reference in New Issue
Block a user