From 6a0041d19c0368b3a1bcc3bb6af33a4df0a72948 Mon Sep 17 00:00:00 2001 From: Michal Piszczek Date: Fri, 3 Sep 2021 15:49:25 +0200 Subject: [PATCH] Enabled postponed evaluation of annotations --- mypy.ini | 2 +- python/tvm/relay/frontend/onnx.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/mypy.ini b/mypy.ini index 17d542c4cb825..f5f81f0acb44f 100644 --- a/mypy.ini +++ b/mypy.ini @@ -32,7 +32,7 @@ ignore_errors = True [mypy-python.tvm.tir.schedule.*] ignore_errors = False -[mypy-python.tvm.relay.frontend.onnx] +[mypy-python.tvm.relay.frontend.onnx.*] ignore_missing_imports = True ignore_errors = True diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index ff8787a06b004..2071bd1469498 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -17,6 +17,8 @@ # pylint: disable=invalid-name, import-self, len-as-condition, unused-argument, too-many-lines # pylint: disable=import-outside-toplevel """ONNX: Open Neural Network Exchange frontend for Relay.""" +from __future__ import __annotations__ + import copy import warnings import typing