tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Tests cleaning & simplification (#81)

Browse Source
This commit is contained in:
Simon Alibert
2024-04-18 14:47:42 +02:00
committed by GitHub
parent 0928afd37d
commit 7ad1909641
24 changed files with 277 additions and 157 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/policies/factory.py
View File

@@ -2,7 +2,7 @@ import inspect
from omegaconf import DictConfig, OmegaConf
from lerobot.common.utils import get_safe_torch_device
from lerobot.common.utils.utils import get_safe_torch_device
def _policy_cfg_from_hydra_cfg(policy_cfg_class, hydra_cfg):

2
lerobot/common/policies/tdmpc/policy.py
View File

@@ -11,7 +11,7 @@ import torch.nn as nn
import lerobot.common.policies.tdmpc.helper as h
from lerobot.common.policies.utils import populate_queues
from lerobot.common.utils import get_safe_torch_device
from lerobot.common.utils.utils import get_safe_torch_device
FIRST_FRAME = 0