dimension_match

DessimozLab · Aug 30, 2024 · 5cdf2d2 · 5cdf2d2
1 parent 1ac0469
commit 5cdf2d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/HogProf/utils/hashutils.py b/src/HogProf/utils/hashutils.py
@@ -66,7 +66,7 @@ def hash_tree(tp , taxaIndex , treeweights , wmg , lossonly = False , duplonly =
             if len(indices[event])>0:
 
                 taxindex = np.asarray(indices[event])
-                hogindex = np.asarray(indices[event])+i*len(taxaIndex)
+                hogindex = np.asarray(indices[event])+i*taxaIndex_max
                 hog_matrix_weighted[:,hogindex] = treeweights[hogindex,:].ravel()
                 if lossonly == True and event == 'loss':
                     hog_matrix_weighted[:,hogindex] = 1