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

Upgrade gym to gymnasium

Browse Source
This commit is contained in:
Simon Alibert
2024-03-25 15:12:21 +01:00
parent 5ef813ff1e
commit dc89166bee
10 changed files with 11 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/common/envs/simxarm/env.py
View File

@@ -20,7 +20,6 @@ from lerobot.common.utils import set_seed
MAX_NUM_ACTIONS = 4
_has_gym = importlib.util.find_spec("gymnasium") is not None
# _has_simxarm = importlib.util.find_spec("simxarm") is not None and _has_gym
class SimxarmEnv(AbstractEnv):

4
lerobot/common/envs/simxarm/simxarm/__init__.py
View File

@@ -1,8 +1,8 @@
from collections import OrderedDict, deque
import gym
import gymnasium as gym
import numpy as np
from gym.wrappers import TimeLimit
from gymnasium.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