Allow index to be any kind of integer

salilab · Dec 20, 2024 · 95a70e1 · 95a70e1
1 parent 836317f
commit 95a70e1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyext/src/tools.py b/pyext/src/tools.py
@@ -18,7 +18,7 @@
 import IMP.rmf
 from collections import defaultdict, OrderedDict
 import warnings
-import numpy
+import numbers
 
 
 def _get_system_for_hier(hier):
@@ -637,7 +637,7 @@ def __init__(self, index):
 
     def add(self, index):
         '''index can be a integer or a list of integers '''
-        if isinstance(index, (int, numpy.int32, numpy.int64)):
+        if isinstance(index, numbers.Integral):
             mergeleft = None
             mergeright = None
             for n, s in enumerate(self.segs):