Merge pull request #16 from wenhao-gao/issue-15

fix: add positional arg
wenhao-gao · Sep 14, 2022 · 658628d · 658628d
2 parents 56917a6 + bf2aaac
commit 658628d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/syn_net/utils/predict_utils.py b/syn_net/utils/predict_utils.py
@@ -868,7 +868,7 @@ def synthetic_tree_decoder_rt1(z_target,
         elif act == 0:
             # Add
             if mol_recent is not None:
-                dist, ind = nn_search(z_mol1)
+                dist, ind = nn_search(z_mol1, _tree=kdtree)
                 mol1 = building_blocks[ind]
             else:
                 dist, ind = nn_search_rt1(z_mol1, _tree=kdtree, _k=rt1_index+1)