[Fix] fix eval_habitat import (#129)

Author: kew6688

internnav/env/utils/habitat_extensions/measures.py

@@ -1,20 +1,13 @@
 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
 from habitat.core.utils import try_cv2_import
 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."""

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 *