fix __init__ import Base

This commit is contained in:
Simon Alibert
2024-03-24 17:57:01 +01:00
parent 1c24bbda3f
commit 0c41675986
2 changed files with 2 additions and 1 deletions

View File

@@ -4,6 +4,7 @@ import gym
import numpy as np
from gym.wrappers import TimeLimit
from lerobot.common.envs.simxarm.simxarm.task.base import Base as Base
from lerobot.common.envs.simxarm.simxarm.task.lift import Lift
from lerobot.common.envs.simxarm.simxarm.task.peg_in_box import PegInBox
from lerobot.common.envs.simxarm.simxarm.task.push import Push