fix: mypy issues

tokoko · Aug 28, 2024 · f855234 · f855234
1 parent 3d12921
commit f855234
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ibis_substrait/compiler/translate.py b/ibis_substrait/compiler/translate.py
@@ -818,11 +818,11 @@ def filter(
 def apply_projection(
     schema_len: int,
     relation: stalg.Rel,
-    values,
+    values: Mapping[str, ops.Value],
     compiler: SubstraitCompiler,
-    child_rel_field_offsets,
-    kwargs,
-):
+    child_rel_field_offsets: Mapping[ops.TableNode, int] | None,
+    kwargs: Mapping,
+) -> stalg.ReadRel:
     mapping_counter = itertools.count(schema_len)
 
     return stalg.Rel(