From 33de6ffa6ecf6fa60df72e05067104825df0572b Mon Sep 17 00:00:00 2001
From: wangyukai <kewa990809@gmail.com>
Date: Tue, 28 Oct 2025 04:20:21 +0000
Subject: [PATCH] fix eval_habitat import

---
 .../env/utils/habitat_extensions/measures.py  | 24 +++++++++++++------
 scripts/eval/eval_habitat.py                  |  2 ++
 2 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/internnav/env/utils/habitat_extensions/measures.py b/internnav/env/utils/habitat_extensions/measures.py
index 751bbf00..5dddcae0 100644
--- a/internnav/env/utils/habitat_extensions/measures.py
+++ b/internnav/env/utils/habitat_extensions/measures.py
@@ -1,10 +1,6 @@
 from typing import Any, List, Union
 
 import numpy as np
-
-# from dtw import dtw
-# from fastdtw import fastdtw
-# from habitat.config import Config
 from habitat.core.embodied_task import EmbodiedTask, Measure
 from habitat.core.registry import registry
 from habitat.core.simulator import Simulator
@@ -12,9 +8,6 @@
 from habitat.tasks.nav.nav import DistanceToGoal
 from numpy import ndarray
 
-# from utils import maps
-# from habitat_extensions.task import RxRVLNCEDatasetV1
-
 cv2 = try_cv2_import()
 
 
@@ -138,6 +131,23 @@ def update_metric(self, *args: Any, **kwargs: Any):
         self._metric += 1.0
 
 
+# import gzip
+# import json
+# import pickle
+# from dtw import dtw
+# from fastdtw import fastdtw
+# from habitat.config import Config
+# from utils import maps
+# from habitat_extensions.task import RxRVLNCEDatasetV1
+# from habitat.tasks.nav.nav import DistanceToGoal, Success
+# from habitat.tasks.utils import cartesian_to_polar
+# from habitat.utils.geometry_utils import quaternion_rotate_vector
+# from habitat.utils.visualizations import fog_of_war
+# from habitat.utils.visualizations import maps as habitat_maps
+# from habitat.core.dataset import Episode
+# from habitat.core.embodied_task import Action, EmbodiedTask, Measure
+# from habitat.core.logging import logger
+
 # @registry.register_measure
 # class WaypointRewardMeasure(Measure):
 #     """A reward measure used for training VLN-CE agents via RL."""
diff --git a/scripts/eval/eval_habitat.py b/scripts/eval/eval_habitat.py
index 2fc420e4..27bf04a3 100644
--- a/scripts/eval/eval_habitat.py
+++ b/scripts/eval/eval_habitat.py
@@ -6,6 +6,8 @@
 import torch
 from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration
 
+# Import for Habitat registry side effects — do not remove
+import internnav.env.utils.habitat_extensions.measures  # noqa: F401
 from internnav.evaluator.habitat_vln_evaluator import VLNEvaluator
 from internnav.model.basemodel.internvla_n1.internvla_n1 import InternVLAN1ForCausalLM
 from internnav.utils.dist import *