From b54a7ab9f362cb1bced9f16e37bb20e681d0c70e Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Wed, 7 Dec 2022 13:07:42 -0500 Subject: [PATCH] Update test/onnx/test_pytorch_onnx_no_runtime.py Co-authored-by: Thiago Crepaldi --- test/onnx/test_pytorch_onnx_no_runtime.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/onnx/test_pytorch_onnx_no_runtime.py b/test/onnx/test_pytorch_onnx_no_runtime.py index 0d0fbed765d3..8a1b959db570 100644 --- 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