[Bugfix] fix import bug, torch is not always installed (#1106)

Co-authored-by: vincentpierre <vincentpierre@users.noreply.github.com>
facebookresearch · Jan 30, 2023 · fe43595 · fe43595
1 parent 84fa782
commit fe43595
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/habitat-lab/habitat/core/vector_env.py b/habitat-lab/habitat/core/vector_env.py
@@ -569,7 +569,7 @@ def _valid_start_methods(self) -> Set[str]:
 
     def _warn_cuda_tensors(
         self,
-        action: Union[int, np.ndarray, Dict[str, Any], torch.Tensor],
+        action: Union[int, np.ndarray, Dict[str, Any], "torch.Tensor"],
         prefix: Optional[str] = None,
     ):
         if torch is None: