From c8c58821f4a3f1e50f9910b550ff4e76fb7de50d Mon Sep 17 00:00:00 2001
From: hyunp2 <42776897+hyunp2@users.noreply.github.com>
Date: Wed, 8 Nov 2023 08:47:44 -0600
Subject: [PATCH] Update generator.py

---
 mofa/generator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mofa/generator.py b/mofa/generator.py
index 75add995..b904050a 100644
--- a/mofa/generator.py
+++ b/mofa/generator.py
@@ -3,7 +3,7 @@
 
 import ase
 
-from model import MOFRecord
+from mofa.model import MOFRecord
 from difflinker_sample import sample_from_sdf
 from difflinker_train import get_args, main
 import yaml