[bug] Support indexing via np.integer

python/taichi/lang/field.py

@@ -352,8 +352,9 @@ def __getitem__(self, key):
         # Check for potential slicing behaviour
         # for instance: x[0, :]
         padded_key = self._pad_key(key)
+        import numpy as np  # pylint: disable=C0415
         for key in padded_key:
-            if not isinstance(key, int):
+            if not isinstance(key, (int, np.integer)):
                 raise TypeError(
                     f"Detected illegal element of type: {type(key)}. "
                     f"Please be aware that slicing a ti.field is not supported so far."

tests/python/test_field.py

@@ -2,6 +2,7 @@
 To test our new `ti.field` API is functional (#1500)
 '''
 
+import numpy as np
 import pytest
 from taichi.lang import impl
 from taichi.lang.misc import get_host_arch_list
@@ -282,6 +283,13 @@ def test_invalid_slicing():
         val[0, :]
 
 
+@test_utils.test()
+def test_slicing_with_np_int():
+    val = ti.field(ti.i32, shape=(2))
+    idx = np.int32(0)
+    val[idx]
+
+
 @test_utils.test(exclude=[ti.cc], debug=True)
 def test_field_fill():
     x = ti.field(int, shape=(3, 3))