(bugfix): Fix mypy typing error + precommit issue

examples/tutorials/nb_python/Habitat2_Quickstart.py

@@ -13,7 +13,7 @@
 #       extension: .py
 #       format_name: percent
 #       format_version: '1.3'
-#       jupytext_version: 1.13.6
+#       jupytext_version: 1.13.8
 #   kernelspec:
 #     display_name: Python 3
 #     name: python3

examples/tutorials/nb_python/habitat2_gym_tutorial.py

@@ -13,7 +13,7 @@
 #       extension: .py
 #       format_name: percent
 #       format_version: '1.3'
-#       jupytext_version: 1.13.6
+#       jupytext_version: 1.13.8
 #   kernelspec:
 #     display_name: Python 3
 #     name: python3

habitat/core/simulator.py

@@ -6,7 +6,16 @@
 import abc
 from collections import OrderedDict
 from enum import Enum
-from typing import Any, Dict, Iterable, List, Optional, Sequence, Union
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Dict,
+    Iterable,
+    List,
+    Optional,
+    Sequence,
+    Union,
+)
 
 import attr
 import numpy as np
@@ -16,7 +25,13 @@
 from habitat.config import Config
 from habitat.core.dataset import Episode
 
-VisualObservation = Union[np.ndarray]
+if TYPE_CHECKING:
+    try:
+        from torch import Tensor
+    except ImportError:
+        pass
+
+VisualObservation = Union[np.ndarray, "Tensor"]
 
 
 @attr.s(auto_attribs=True)

habitat/sims/habitat_simulator/habitat_simulator.py

@@ -234,7 +234,9 @@ class HabitatSimFisheyeSemanticSensor(HabitatSimSemanticSensor):
     _get_default_spec = habitat_sim.FisheyeSensorDoubleSphereSpec
 
 
-def check_sim_obs(obs: Optional[np.ndarray], sensor: Sensor) -> None:
+def check_sim_obs(
+    obs: Union[np.ndarray, "Tensor", None], sensor: Sensor
+) -> None:
     assert obs is not None, (
         "Observation corresponding to {} not present in "
         "simulator's observations".format(sensor.uuid)