⭐ fix to be able to use recent numpy

DwangoMediaVillage · Oct 14, 2023 · ac2960d · ac2960d
1 parent ebaceca
commit ac2960d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pqkmeans/encoder/pq_encoder.py b/pqkmeans/encoder/pq_encoder.py
@@ -22,9 +22,9 @@ def fit(self, x_train):
         self.Ds = int(D / self.M)
         assert self.trained_encoder is None, "fit must be called only once"
 
-        codewords = numpy.zeros((self.M, self.Ks, self.Ds), dtype=numpy.float)
+        codewords = numpy.zeros((self.M, self.Ks, self.Ds), dtype=float)
         for m in range(self.M):
-            x_train_sub = x_train[:, m * self.Ds: (m + 1) * self.Ds].astype(numpy.float)
+            x_train_sub = x_train[:, m * self.Ds: (m + 1) * self.Ds].astype(float)
             codewords[m], _ = kmeans2(x_train_sub, self.Ks, iter=self.iteration, minit='points')
         self.trained_encoder = TrainedPQEncoder(codewords, self.code_dtype)
 
@@ -66,7 +66,7 @@ def decode_multi(self, codes):
         assert M == self.M
         assert codes.dtype == self.code_dtype
 
-        decoded = numpy.empty((N, self.Ds * self.M), dtype=numpy.float)
+        decoded = numpy.empty((N, self.Ds * self.M), dtype=float)
         for m in range(self.M):
             decoded[:, m * self.Ds: (m + 1) * self.Ds] = self.codewords[m][codes[:, m], :]
         return decoded
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-numpy<1.21
+numpy
 scikit-learn
 pipe<2.0
 scipy