From 3f2991be6ddca63b342a63840351bc66df897cab Mon Sep 17 00:00:00 2001
From: hyunp2 <42776897+hyunp2@users.noreply.github.com>
Date: Tue, 26 Sep 2023 21:16:13 -0500
Subject: [PATCH] Update 3_difflinker_sample.py

---
 3_difflinker_sample.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/3_difflinker_sample.py b/3_difflinker_sample.py
index 71d7a9c4..b2e81e4f 100644
--- a/3_difflinker_sample.py
+++ b/3_difflinker_sample.py
@@ -17,7 +17,7 @@
             OUTPUT_DIR = f'output/n_atoms_{n_atoms}/{node}'
             os.makedirs(OUTPUT_DIR,exist_ok=True)
             # subprocess.run(f'python -W ignore utils/difflinker_sample_and_analyze.py --linker_size {n_atoms} --fragments data/fragments_all/{node}/hMOF_frag.sdf --model models/geom_difflinker.ckpt --output {OUTPUT_DIR} --n_samples 1',shell=True)
-            main_run(input_path="data/fragments_all/{node}/hMOF_frag_frag.sdf", model="models/geom_difflinker.ckpt", linker_size=str(n_atoms), output_dir=OUTPUT_DIR, n_samples=3, n_steps=None, anchors=None)
+            main_run(input_path=f"data/fragments_all/{node}/hMOF_frag_frag.sdf", model="models/geom_difflinker.ckpt", linker_size=str(n_atoms), output_dir=OUTPUT_DIR, n_samples=1, n_steps=None, anchors=None)
 
             # change to the line below to reproduce paper result
             #subprocess.run(f'python -W ignore utils/difflinker_sample_and_analyze.py --linker_size {n_atoms} --fragments data/fragments_all/{node}/hMOF_frag.sdf --model models/geom_difflinker.ckpt --output {OUTPUT_DIR} --n_samples 20',shell=True)