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

fix import

Browse Source
This commit is contained in:
Cadene
2024-03-11 12:41:14 +00:00
parent ccd5dc5a42
commit 84a1647c01
2 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/common/envs/factory.py
View File

@@ -1,6 +1,4 @@
from torchrl.envs.transforms import StepCounter, TransformedEnv
from rl.torchrl.envs.transforms.transforms import Compose, Transform
from torchrl.envs.transforms import Compose, StepCounter, Transform, TransformedEnv
def make_env(cfg, transform=None):