use np sum

biocore · Jun 26, 2024 · 06c4ac4 · 06c4ac4
1 parent ca7ac23
commit 06c4ac4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/bp/_conv.pyx b/bp/_conv.pyx
@@ -24,9 +24,9 @@ def to_skbio_treenode(BP bp):
         The tree represented as an skbio.TreeNode
     """
     cdef int i
-
-    nodes = [skbio.TreeNode() for i in range(sum(bp.B))]
-
+    
+    nodes = [skbio.TreeNode() for i in range(bp.B.sum())]
+    
     # skbio.TreeNode.append makes a very expensive call to
     # invalidate_caches. Let's remove that from consideration
     # temporarily while constructing the tree
@@ -41,7 +41,7 @@ def to_skbio_treenode(BP bp):
 
     root = nodes[0]
 
-    for i in range(sum(bp.B)):
+    for i in range(bp.B.sum()):
         node_idx = bp.preorderselect(i)
         nodes[i].name = bp.name(node_idx)
         nodes[i].length = bp.length(node_idx)

diff --git a/setup.py b/setup.py
@@ -101,7 +101,8 @@ def run(self):
 
 if USE_CYTHON:
     from Cython.Build import cythonize
-    extensions = cythonize(extensions)
+    extensions = cythonize(extensions,
+                           language_level=3)
 
 
 setup(name='iow',