Update test/onnx/test_pytorch_onnx_no_runtime.py

Co-authored-by: Thiago Crepaldi <thiago.crepaldi@microsoft.com>
pytorch · Dec 7, 2022 · b54a7ab · b54a7ab
1 parent b265e25
commit b54a7ab
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/onnx/test_pytorch_onnx_no_runtime.py b/test/onnx/test_pytorch_onnx_no_runtime.py
@@ -15,6 +15,7 @@
 import torch
 import torch.nn.functional as F
 from torch import Tensor
+import numpy as np
 from torch.onnx import OperatorExportTypes, symbolic_helper, utils
 from torch.onnx._globals import GLOBALS
 from torch.onnx._internal import registration