Fixed issue #4

RapidAI · Dec 13, 2023 · 50eef11 · 50eef11
1 parent 2f905a5
commit 50eef11
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lineless_table_rec/lineless_table_process.py b/lineless_table_rec/lineless_table_process.py
@@ -273,7 +273,7 @@ def _gather(data: np.ndarray, dim: int, index: np.ndarray) -> np.ndarray:
             + ", all dimensions of index and data should be the same size"
         )
 
-    if index.dtype != np.dtype("int_"):
+    if index.dtype != np.int64:
         raise TypeError("The values of index must be integers")
 
     data_swaped = np.swapaxes(data, 0, dim)

diff --git a/wired_table_rec/utils_table_line_rec.py b/wired_table_rec/utils_table_line_rec.py
@@ -156,7 +156,7 @@ def _gather_np(data, dim, index):
             + ", all dimensions of index and data should be the same size"
         )
 
-    if index.dtype != np.dtype("int_"):
+    if index.dtype != np.int64:
         raise TypeError("The values of index must be integers")
 
     data_swaped = np.swapaxes(data, 0, dim)