diff --git a/tests/policies/molmoact2/test_molmoact2.py b/tests/policies/molmoact2/test_molmoact2.py index 7e70ce4f6..b2c6cbfca 100644 --- a/tests/policies/molmoact2/test_molmoact2.py +++ b/tests/policies/molmoact2/test_molmoact2.py @@ -14,6 +14,10 @@ # See the License for the specific language governing permissions and # limitations under the License. +"""Unit tests for MolmoAct2's LeRobot policy interface.""" + +# ruff: noqa: E402 + from __future__ import annotations import json @@ -25,6 +29,9 @@ import pytest import torch import torch.nn.functional as F # noqa: N812 +pytest.importorskip("transformers") +pytest.importorskip("scipy") + from lerobot.configs import FeatureType, NormalizationMode, PolicyFeature from lerobot.policies import get_policy_class, make_policy_config from lerobot.policies.molmoact2 import (