Upload ENVmodel.py
Browse files- ENVmodel.py +5 -4
ENVmodel.py
CHANGED
|
@@ -3,6 +3,10 @@
|
|
| 3 |
|
| 4 |
import os
|
| 5 |
import numpy as np
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
|
| 7 |
os.environ["MUJOCO_GL"] = "egl"
|
| 8 |
|
|
@@ -93,10 +97,7 @@ class OffscreenRenderer:
|
|
| 93 |
|
| 94 |
"""Gymnasium environment for SO-101 pick-and-place in MuJoCo."""
|
| 95 |
|
| 96 |
-
|
| 97 |
-
import mujoco
|
| 98 |
-
import numpy as np
|
| 99 |
-
from gymnasium import spaces
|
| 100 |
|
| 101 |
class SO101PickPlaceEnv(gym.Env):
|
| 102 |
|
|
|
|
| 3 |
|
| 4 |
import os
|
| 5 |
import numpy as np
|
| 6 |
+
import gymnasium as gym
|
| 7 |
+
import mujoco
|
| 8 |
+
import numpy as np
|
| 9 |
+
from gymnasium import spaces
|
| 10 |
|
| 11 |
os.environ["MUJOCO_GL"] = "egl"
|
| 12 |
|
|
|
|
| 97 |
|
| 98 |
"""Gymnasium environment for SO-101 pick-and-place in MuJoCo."""
|
| 99 |
|
| 100 |
+
|
|
|
|
|
|
|
|
|
|
| 101 |
|
| 102 |
class SO101PickPlaceEnv(gym.Env):
|
| 103 |
|