fix ValueError: Buffer dtype mismatch, expected 'int_t' but got 'long…

… long' at self.min_cost = compute(self.n, self.c, self._x, self._y, self.epsilon)
materialsproject · Jan 9, 2023 · 856bfb5 · 856bfb5
1 parent ce9fe2e
commit 856bfb5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pymatgen/optimization/linear_assignment.pyx b/pymatgen/optimization/linear_assignment.pyx
@@ -64,8 +64,8 @@ class LinearAssignment:
             self.c[:self.nx] = self.orig_c
 
         # initialize solution vectors
-        self._x = np.empty(self.n, dtype=np.int64)
-        self._y = np.empty(self.n, dtype=np.int64)
+        self._x = np.empty(self.n, dtype=int)
+        self._y = np.empty(self.n, dtype=int)
 
         self.min_cost = compute(self.n, self.c, self._x, self._y, self.epsilon)
         self.solution = self._x[:self.nx]