swpark5 commited on
Commit
f355fb3
·
1 Parent(s): 0aef937

Revert: 수동 다운로드를 제거하고 상대 경로를 사용

Browse files
Files changed (1) hide show
  1. env.py +5 -25
env.py CHANGED
@@ -2,33 +2,13 @@ import gymnasium as gym
2
  import mujoco
3
  import numpy as np
4
  import os
5
- from huggingface_hub import hf_hub_download # 1. Hub 다운로더 임포트
6
 
7
- # --- 2. 다운로드할 파일 정보 정의 ---
8
- # 이 환경 레포의 ID
9
- ENV_REPO_ID = "swpark5/so101-sim-env"
10
 
11
- # (중요!) Hub 레포에 있는 정확한 파일 경로
12
- # (이전 git status 로그 기준)
13
- MODEL_FILENAME = "SO-ARM100/Simulation/SO101/scene.xml"
14
- # <include file="so101_new_calib.xml" /> 파일
15
- CALIB_FILENAME = "SO-ARM100/Simulation/SO101/so101_new_calib.xml"
16
-
17
- try:
18
- # --- 3. env.py 실행 시 모델 파일들을 수동으로 다운로드 ---
19
-
20
- # scene.xml 다운로드 (절대 경로 반환)
21
- MODEL_PATH = hf_hub_download(repo_id=ENV_REPO_ID, filename=MODEL_FILENAME)
22
-
23
- # scene.xml이 <include>하는 calib.xml 파일도 다운로드
24
- # (이걸 해줘야 MuJoCo가 로드할 때 파일을 찾을 수 있음)
25
- hf_hub_download(repo_id=ENV_REPO_ID, filename=CALIB_FILENAME)
26
-
27
- except Exception as e:
28
- print(f"--- Hugging Face Hub에서 모델 파일({MODEL_FILENAME}) 다운로드 실패 ---")
29
- print(f"레포 ID({ENV_REPO_ID})와 파일 경로가 올바른지 확인하세요.")
30
- print(f"원본 오류: {e}")
31
- raise
32
 
33
 
34
  class SO101SimEnv(gym.Env):
 
2
  import mujoco
3
  import numpy as np
4
  import os
 
5
 
6
+ # --- 1. SO-101 커스텀 Gym 환경 (우리가 만들었던 것) ---
 
 
7
 
8
+ # 이 파일(env.py) 기준 상대 경로
9
+ MODEL_PATH = os.path.join(
10
+ os.path.dirname(__file__), "SO-ARM100", "Simulation", "SO101", "scene.xml"
11
+ )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
 
13
 
14
  class SO101SimEnv(gym.Env):