diff --git a/lineless_table_rec/lineless_table_process.py b/lineless_table_rec/lineless_table_process.py index 3b9b7f9..7043d3c 100644 --- 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 index ef6afff..0e2caa5 100644 --- 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)