Fix premap_ids

explosion · Sep 2, 2024 · e67b917 · e67b917
1 parent 0900b51
commit e67b917
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/thinc/layers/premap_ids.pyx b/thinc/layers/premap_ids.pyx
@@ -36,9 +36,9 @@ cdef lookup(PreshMap mapping, int64_t[:] keys, int64_t default):
 @registry.layers("premap_ids.v1")
 def premap_ids(
     mapping_table: Mapping[int, int],
-    default: int32_t = 0,
+    default: int = 0,
     *,
-    column: Optional[int32_t] = None
+    column: Optional[int] = None
 ):
     """Remap integer inputs to integers a mapping table, usually as a
     preprocess before embeddings."""