.

mlc-ai · Jan 14, 2023 · aee20f0 · aee20f0
1 parent 2a53757
commit aee20f0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/mypy.ini b/mypy.ini
@@ -30,7 +30,4 @@ ignore_errors = True
 ignore_errors = True
 
 [mypy-python.tvm.tir.schedule.*]
-ignore_errors = False
-
-[mypy-python.tvm.relax.*]
-ignore_missing_imports = True
+ignore_errors = False
diff --git a/python/tvm/relax/frontend/pytorch_fx.py b/python/tvm/relax/frontend/pytorch_fx.py
@@ -20,9 +20,9 @@
 from typing import Dict
 
 from tvm import relax
-import torch
+import torch  # type: ignore
 from torch import nn, fx
-import numpy as np
+import numpy as np  # type: ignore
 
 
 class TorchFXTranslator: