nepyope commited on
Commit
ec54ae0
·
verified ·
1 Parent(s): ef6a683

Update env.py

Browse files
Files changed (1) hide show
  1. env.py +7 -2
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)