update dtype of numpy

fangwei123456 · Jun 2, 2023 · a5a51db · a5a51db
1 parent b44d109
commit a5a51db
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/spikingjelly/activation_based/auto_cuda/base.py b/spikingjelly/activation_based/auto_cuda/base.py
@@ -226,7 +226,7 @@ def check_ctypes(self, py_dict: dict):
 
         ``torch.half`` or ``np.float16`` ------ ``'const half2'`` or ``'half2'``
 
-        ``np.int``  ------  ``'const int'`` or ``'int'``
+        ``np.int_``  ------  ``'const int'`` or ``'int'``
 
         If not, this function will raise an error.
         """
@@ -246,7 +246,7 @@ def check_ctypes(self, py_dict: dict):
                 elif value.dtype == np.float16:
                     assert startswiths(ctype, ('const half2', 'half2'))
 
-                elif value.dtype == np.int:
+                elif value.dtype == np.int_:
                     assert startswiths(ctype, ('const int', 'int'))
 
     def check_half2(self, py_dict: dict):

diff --git a/spikingjelly/datasets/cifar10_dvs.py b/spikingjelly/datasets/cifar10_dvs.py
@@ -83,7 +83,7 @@ def parse_raw_address(addr,
                       y_shift=y_shift,
                       polarity_mask=polarity_mask,
                       polarity_shift=polarity_shift):
-    polarity = read_bits(addr, polarity_mask, polarity_shift).astype(np.bool)
+    polarity = read_bits(addr, polarity_mask, polarity_shift).astype(np.bool_)
     x = read_bits(addr, x_mask, x_shift)
     y = read_bits(addr, y_mask, y_shift)
     return x, y, polarity