2024-02-25 18:50:23 +08:00
|
|
|
import pytest
|
|
|
|
|
|
|
|
|
|
from lerobot.common.policies.factory import make_policy
|
|
|
|
|
|
|
|
|
|
from .utils import init_config
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.parametrize(
|
2024-02-26 01:42:47 +08:00
|
|
|
"env_name",
|
2024-02-25 18:50:23 +08:00
|
|
|
[
|
2024-02-26 01:42:47 +08:00
|
|
|
"simxarm",
|
2024-02-25 18:50:23 +08:00
|
|
|
"pusht",
|
|
|
|
|
],
|
|
|
|
|
)
|
2024-02-26 01:42:47 +08:00
|
|
|
def test_factory(env_name):
|
|
|
|
|
cfg = init_config(overrides=[f"env={env_name}"])
|
2024-02-25 18:50:23 +08:00
|
|
|
policy = make_policy(cfg)
|
