diff --git a/src/HogProf/utils/hashutils.py b/src/HogProf/utils/hashutils.py index 1679a1c..f5dd62a 100755 --- a/src/HogProf/utils/hashutils.py +++ b/src/HogProf/utils/hashutils.py @@ -55,8 +55,8 @@ def hash_tree(tp , taxaIndex , treeweights , wmg , lossonly = False , duplonly = return None, None taxaIndex_max = max(taxaIndex.values())+1 - hog_matrix_weighted = np.zeros((1, 3*len(taxaIndex_max))) - hog_matrix_binary = np.zeros((1, 3*len(taxaIndex_max))) + hog_matrix_weighted = np.zeros((1, 3*taxaIndex_max)) + hog_matrix_binary = np.zeros((1, 3*taxaIndex_max)) if tp: losses = [ taxaIndex[n.name] for n in tp.traverse() if n.lost and n.name in taxaIndex ] dupl = [ taxaIndex[n.name] for n in tp.traverse() if n.dupl and n.name in taxaIndex ]