From a0bd1bf8f2bf50bb0632cfe9462f74056ca232f3 Mon Sep 17 00:00:00 2001
From: hyunp2 <42776897+hyunp2@users.noreply.github.com>
Date: Tue, 17 Oct 2023 23:45:36 -0500
Subject: [PATCH] Update generator.py

---
 mofa/generator.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/mofa/generator.py b/mofa/generator.py
index cd78af4a..00dac2b7 100644
--- a/mofa/generator.py
+++ b/mofa/generator.py
@@ -4,8 +4,7 @@
 import ase
 
 from model import MOFRecord
-from difflinker_sample import 
-
+from difflinker_sample import sampler
 
 def train_generator(
         starting_model: str | Path,
@@ -37,4 +36,6 @@ def run_generator(
     Returns:
         3D geometries of the generated linkers
     """
+    sampler(nodes: List[str]=['CuCu'], n_atoms_list)
+        
     raise NotImplementedError()