deploy/go1_env.py

#!/usr/bin/env python3
"""Portable path config for the G2 / GO-1 deploy scripts.

所有"宿主机/机器人相关"的路径都从环境变量读取，默认值按本仓库的目录结构推导，
因此把仓库 checkout 到任何位置都能直接用，无需改代码。可用环境变量覆盖：

  GO1_CKPT          模型 checkpoint 目录   (默认: <repo>/checkpoint_22000)
  GO1_WORKSPACE     GDK / go2_data_recorder 安装位置 (默认: /home/agi/workspace)
  GO1_AGIBOT_WORLD  AgiBot-World 模型代码   (默认: /data/agi/AgiBot-World)
  GO1_POSE_FILE     保存的初始位姿 .npy     (默认: <deploy>/saved_init_pose.npy)
  GO1_OUT_DIR       grab_*.jpg 等输出目录   (默认: <deploy>)

`import go1_env` 时会把 GO1_AGIBOT_WORLD 和 GO1_WORKSPACE 插入 sys.path
（在 `import agibot_gdk` / 模型模块之前需要）。
"""
import os
import sys

_HERE = os.path.dirname(os.path.abspath(__file__))      # .../deploy
_REPO = os.path.dirname(_HERE)                          # 仓库根 (checkpoint_22000 的父目录)

CKPT = os.environ.get("GO1_CKPT", os.path.join(_REPO, "checkpoint_22000"))
STATS = os.path.join(CKPT, "dataset_stats.json")
WORKSPACE = os.environ.get("GO1_WORKSPACE", "/home/agi/workspace")
AGIBOT_WORLD = os.environ.get("GO1_AGIBOT_WORLD", "/data/agi/AgiBot-World")
POSE_FILE = os.environ.get("GO1_POSE_FILE", os.path.join(_HERE, "saved_init_pose.npy"))
OUT_DIR = os.environ.get("GO1_OUT_DIR", _HERE)

for _p in (AGIBOT_WORLD, WORKSPACE):
    if _p and _p not in sys.path:
        sys.path.insert(0, _p)