ty fixes

Author: rustyconover

pyproject.toml

@@ -42,6 +42,11 @@ warn_unused_ignores = true
 module = "structlog.*"
 ignore_missing_imports = true
 
+[[tool.mypy.overrides]]
+# These files have type: ignore comments for ty that mypy doesn't need
+module = ["vgi.examples.scalar", "vgi.table_function"]
+warn_unused_ignores = false
+
 [tool.ty.environment]
 python-version = "3.12"
 

vgi/examples/scalar.py

@@ -52,7 +52,7 @@ def output_type(self) -> pa.DataType:
 
     def compute(self, batch: pa.RecordBatch) -> pa.Array[Any]:
         """Double the values in the specified column."""
-        return pc.multiply(batch.column(self.column), 2)
+        return pc.multiply(batch.column(self.column), 2)  # type: ignore[no-matching-overload]
 
 
 class AddColumnsFunction(ScalarFunction):
@@ -109,4 +109,4 @@ def output_type(self) -> pa.DataType:
 
     def compute(self, batch: pa.RecordBatch) -> pa.Array[Any]:
         """Convert the column values to uppercase."""
-        return pc.utf8_upper(batch.column(self.column))
+        return pc.utf8_upper(batch.column(self.column))  # type: ignore[no-matching-overload]

vgi/table_function.py

@@ -23,7 +23,7 @@
 
 from collections.abc import Generator
 from dataclasses import dataclass
-from typing import Any, final
+from typing import Any, Self, final
 
 import pyarrow as pa
 import structlog
@@ -331,14 +331,14 @@ def serialize(self) -> bytes:
         return vgi.ipc_utils.serialize_record_batch(batch)
 
     @classmethod
-    def deserialize(cls, batch: pa.RecordBatch) -> "TableFunctionInitInput":
+    def deserialize(cls, batch: pa.RecordBatch) -> Self:  # type: ignore[override]
         """Deserialize TableFunctionInitInput from a RecordBatch."""
         values = batch.to_pylist()[0]
         # Handle backward compatibility: ignore extra fields
         return cls(projection_ids=values.get("projection_ids"))
 
     @classmethod
-    def deserialize_bytes(cls, data: bytes) -> "TableFunctionInitInput":
+    def deserialize_bytes(cls, data: bytes) -> Self:
         """Deserialize TableFunctionInitInput from bytes."""
         batch = vgi.ipc_utils.deserialize_record_batch(data)
         return cls.deserialize(batch)