fix(actpred): gen_calibration --deploy 的 is_range_mode 未导入即使用

import 在 if 分支内滞后于首次调用,fixed_range 归一化模型生成校准集
时直接 NameError;导入提前到 build_deploy_set 开头。
This commit is contained in:
qing committed 2026-08-16 16:17:10 +00:00
1 parent d8b601ab9a
commit 293d71e86b
1 file changed
+1
+1
View File
@@ -86,6 +86,7 @@ def build_image_samples(n_samples, frame_len, height, width, rng,
def build_deploy_set(args):
import torch
from action_norm import is_range_mode, normalize_action
from model import ActionPredictor
ckpt = torch.load(args.model, map_location="cpu", weights_only=True)