Upd premap_ids

explosion · Sep 2, 2024 · 0900b51 · 0900b51
1 parent 33295fb
commit 0900b51
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/thinc/layers/premap_ids.pyx b/thinc/layers/premap_ids.pyx
@@ -35,7 +35,7 @@ cdef lookup(PreshMap mapping, int64_t[:] keys, int64_t default):
 
 @registry.layers("premap_ids.v1")
 def premap_ids(
-    mapping_table: Mapping[int32_t, int32_t],
+    mapping_table: Mapping[int, int],
     default: int32_t = 0,
     *,
     column: Optional[int32_t] = None