Add policies/factory, Add test, Add _self_ in config

This commit is contained in:
Cadene
2024-02-25 10:50:23 +00:00
parent 64b5920e94
commit 598bb496b0
13 changed files with 61 additions and 38 deletions

17
test/test_policies.py Normal file
View File

@@ -0,0 +1,17 @@
import pytest
from lerobot.common.policies.factory import make_policy
from .utils import init_config
@pytest.mark.parametrize(
"config_name",
[
"default",
"pusht",
],
)
def test_factory(config_name):
cfg = init_config(config_name)
policy = make_policy(cfg)