fix typo

open-mmlab · Dec 28, 2020 · d63ca69 · d63ca69
1 parent 2ef9449
commit d63ca69
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/mmcv/utils/parrots_jit.py b/mmcv/utils/parrots_jit.py
@@ -1,6 +1,4 @@
-import torch
-
-TORCH_VERSION = torch.__version__
+from .parrots_wrapper import TORCH_VERSION
 
 if TORCH_VERSION == 'parrots':
     from parrots.jit import pat as jit

diff --git a/mmcv/utils/parrots_wrapper.py b/mmcv/utils/parrots_wrapper.py
@@ -2,7 +2,7 @@
 
 import torch
 
-from .parrots_wrapper import TORCH_VERSION
+TORCH_VERSION = torch.__version__
 
 
 def _get_cuda_home():