more careful casting

dedupeio · Jun 20, 2024 · b43ce75 · b43ce75
1 parent 098129a
commit b43ce75
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/dedupe/convenience.py b/dedupe/convenience.py
@@ -43,18 +43,18 @@ def randomPairs(n_records: int, sample_size: int) -> IndicesIterator:
     else:
         try:
             random_pairs = numpy.array(
-                random.sample(range(n), sample_size), dtype=numpy.uint
+                random.sample(range(n), sample_size), dtype=numpy.uint64
             )
         except OverflowError:
             return randomPairsWithReplacement(n_records, sample_size)
 
     b: int = 1 - 2 * n_records
 
     i = (-b - 2 * numpy.sqrt(2 * (n - random_pairs) + 0.25)) // 2
-    i = i.astype(numpy.uint)
+    i = i.astype(numpy.int64)
 
     j = random_pairs + i * (b + i + 2) // 2 + 1
-    j = j.astype(numpy.uint)
+    j = j.astype(numpy.uint64)
 
     return zip(i, j)
 

diff --git a/dedupe/variables/interaction.py b/dedupe/variables/interaction.py
@@ -11,7 +11,7 @@ class InteractionType(Variable):
     type = "Interaction"
     higher_vars: list[InteractionVariable]
 
-    def __init__(self, *args, **kwargs):
+    def __init__(self, *args: str, **kwargs):
         self.interactions = list(args)
 
         self.name = "(Interaction: %s)" % str(self.interactions)