Fix lint

Author: justinchuby

onnxscript/test/onnx_types_test.py

@@ -6,6 +6,7 @@
 # mypy: disable-error-code=misc
 
 """Unit tests for the onnx_types module."""
+from __future__ import annotations
 
 import unittest
 
@@ -24,13 +25,13 @@ def test_instantiation(self):
             FLOAT[...]()
 
     @parameterized.expand(tensor_type_registry.items())
-    def test_type_properties(self, dtype: DType, tensor_type: TensorType):
+    def test_type_properties(self, dtype: DType, tensor_type: type[TensorType]):
         self.assertEqual(tensor_type.dtype, dtype)
         self.assertIsNone(tensor_type.shape)
-        self.assertEqual(tensor_type[...].shape, ...)
-        self.assertEqual(tensor_type[...].dtype, dtype)
-        self.assertEqual(tensor_type[1, 2, 3].shape, (1, 2, 3))
-        self.assertEqual(tensor_type[1, 2, 3].dtype, dtype)
+        self.assertEqual(tensor_type[...].shape, ...)  # type: ignore[index]
+        self.assertEqual(tensor_type[...].dtype, dtype)  # type: ignore[index]
+        self.assertEqual(tensor_type[1, 2, 3].shape, (1, 2, 3))  # type: ignore[index]
+        self.assertEqual(tensor_type[1, 2, 3].dtype, dtype)  # type: ignore[index]
 
     @parameterized.expand([(dtype,) for dtype in tensor_type_registry])
     def test_dtype_bound_to_subclass(self, dtype: DType):