lint

microsoft · Oct 23, 2024 · e6c36dd · e6c36dd
1 parent c10f5f5
commit e6c36dd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxscript/ir/_schemas_test.py b/onnxscript/ir/_schemas_test.py
@@ -144,7 +144,7 @@ class TypeConversionFunctionsTest(unittest.TestCase):
         ]
     )
     def test_pytype_to_ir_type(self, _, pytype: Any, expected: set[ir.TypeProtocol]):
-        self.assertEqual(_schemas._get_allowed_types_from_type_annotation(pytype), expected)
+        self.assertEqual(_schemas._get_allowed_types_from_type_annotation(pytype), expected)  # pylint: disable=protected-access
 
     @parameterized.parameterized.expand(
         [
@@ -169,7 +169,7 @@ def test_pytype_to_ir_type(self, _, pytype: Any, expected: set[ir.TypeProtocol])
         ]
     )
     def test_get_type_constraint_name(self, _: str, pytype: Any, expected: str | None):
-        self.assertEqual(_schemas._get_type_constraint_name(pytype), expected)
+        self.assertEqual(_schemas._get_type_constraint_name(pytype), expected)  # pylint: disable=protected-access
 
 
 if __name__ == "__main__":