Update env.py
Browse files
env.py
CHANGED
|
@@ -1,6 +1,11 @@
|
|
| 1 |
# env.py
|
|
|
|
|
|
|
| 2 |
from pathlib import Path
|
| 3 |
import yaml
|
|
|
|
|
|
|
|
|
|
| 4 |
|
| 5 |
from sim.simulator_factory import SimulatorFactory, init_channel
|
| 6 |
|
|
@@ -70,9 +75,9 @@ class UnitreeG1SimulationEnvironment:
|
|
| 70 |
self.sim.close()
|
| 71 |
|
| 72 |
|
| 73 |
-
def make_env(**kwargs):
|
| 74 |
"""
|
| 75 |
Required entrypoint for EnvHub.
|
| 76 |
Returns an environment instance.
|
| 77 |
"""
|
| 78 |
-
return UnitreeG1SimulationEnvironment(**kwargs)
|
|
|
|
| 1 |
# env.py
|
| 2 |
+
|
| 3 |
+
|
| 4 |
from pathlib import Path
|
| 5 |
import yaml
|
| 6 |
+
from huggingface_hub import snapshot_download
|
| 7 |
+
import os
|
| 8 |
+
snapshot_download(repo_id='lerobot/unitree-g1-mujoco', revision=None, cache_dir=None)
|
| 9 |
|
| 10 |
from sim.simulator_factory import SimulatorFactory, init_channel
|
| 11 |
|
|
|
|
| 75 |
self.sim.close()
|
| 76 |
|
| 77 |
|
| 78 |
+
def make_env(n_envs=1,use_async_envs: bool = False, **kwargs):
|
| 79 |
"""
|
| 80 |
Required entrypoint for EnvHub.
|
| 81 |
Returns an environment instance.
|
| 82 |
"""
|
| 83 |
+
return UnitreeG1SimulationEnvironment(**kwargs)
|