diff --git a/pyproject.toml b/pyproject.toml index 3721ffed72717c74fb1db0db3b64a942830718d2..810c5d9f64e9cea6ab6f58a6fca8a3950b7a8156 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -107,8 +107,6 @@ ignore = [ "F403", "F632", "F811", - "F821", - "F901", "C408", "C417", "PLR0402", diff --git a/test/cinn/test_paddle_model_convertor.py b/test/cinn/test_paddle_model_convertor.py index 8d8313807a2ff290544f06cb3f044cef53f43a9b..699bb0dc093afc246e06298adaa7bf6992fa0556 100644 --- a/test/cinn/test_paddle_model_convertor.py +++ b/test/cinn/test_paddle_model_convertor.py @@ -21,7 +21,11 @@ import sys import unittest import numpy as np -from cinn.common import DefaultNVGPUTarget, is_compiled_with_cuda +from cinn.common import ( + DefaultHostTarget, + DefaultNVGPUTarget, + is_compiled_with_cuda, +) from cinn.frontend import PaddleModelConvertor from cinn.runtime import seed as cinn_seed from op_mappers.op_mapper_test import OpMapperTest diff --git a/test/cinn/test_utils.py b/test/cinn/test_utils.py index 45dd9ae629945cc35c7ba6ef710a5a6f25eaf1cb..e1a999d40df549837229d9095c08fab6d502a3fd 100755 --- a/test/cinn/test_utils.py +++ b/test/cinn/test_utils.py @@ -43,7 +43,7 @@ class SingleOpTester(unittest.TestCase): ''' create the target of the operator's execution output. ''' - raise NotImplemented + raise NotImplementedError def test_op(self): '''