accept np.ndarray as index (#407)

deepmodeling · Jan 12, 2023 · cfc9fd4 · cfc9fd4
1 parent e6cbaab
commit cfc9fd4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dpdata/system.py b/dpdata/system.py
@@ -339,7 +339,7 @@ def __str__(self):
 
     def __getitem__(self, key):
         """Returns proerty stored in System by key or by idx"""
-        if isinstance(key, (int, slice, list)):
+        if isinstance(key, (int, slice, list, np.ndarray)):
             return self.sub_system(key)
         return self.data[key]