added AHC

GuoJeff · Dec 14, 2022 · 955dd2a · 955dd2a
1 parent 61e5ea4
commit 955dd2a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/run.py b/run.py
@@ -61,6 +61,8 @@ def main():
         from main.hebo.run import HEBO_Optimizer as Optimizer 
     elif args.method == 'graph_mcts':
         from main.graph_mcts.run import Graph_MCTS_Optimizer as Optimizer
+    elif args.method == 'smiles_ahc':
+        from main.smiles_ahc.run import AHC_Optimizer as Optimizer
     elif args.method == "smiles_lstm_hc":
         from main.smiles_lstm_hc.run import SMILES_LSTM_HC_Optimizer as Optimizer
     elif args.method == 'selfies_lstm_hc':