Imports

pytorch · Oct 13, 2022 · e6aeb3a · e6aeb3a
1 parent cc1f552
commit e6aeb3a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/onnx/test_utility_funs.py b/test/onnx/test_utility_funs.py
@@ -4,6 +4,7 @@
 import functools
 import io
 from typing import Callable
+import warnings
 
 import onnx
 import parameterized
@@ -157,8 +158,6 @@ def forward(self, x):
             self.assertFalse(torch.onnx.is_in_onnx_export())
 
     def test_validate_dynamic_axes_invalid_input_output_name(self):
-        import warnings
-
         with warnings.catch_warnings(record=True) as w:
             warnings.simplefilter("always")
             utils._validate_dynamic_axes(