From b81886bcc2e562abb31730336ab8ec6255d83928 Mon Sep 17 00:00:00 2001
From: hyunp2 <42776897+hyunp2@users.noreply.github.com>
Date: Sun, 3 Dec 2023 23:30:30 -0600
Subject: [PATCH] Update fragment.py

---
 mofa/fragment.py | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/mofa/fragment.py b/mofa/fragment.py
index 4f1abcd2..20ce49b2 100644
--- a/mofa/fragment.py
+++ b/mofa/fragment.py
@@ -5,10 +5,15 @@
 
 from model import MOFRecord
 
+from mofa.difflinker_fragmentation import fragmentation
+from mofa.difflinker_process_fragmentation import process_fragments
+from typing import *
+
 def fragment_mof_linkers(
-        starting_model: str | Path,
-        examples: list[MOFRecord],
-        num_epochs: int
+        nodes: List[str] = ["CuCu"]
+        # starting_model: str | Path,
+        # examples: list[MOFRecord],
+        # num_epochs: int
 ) -> Path:
     """Fragment linkers of MOFs
 
@@ -19,5 +24,7 @@ def fragment_mof_linkers(
     Returns:
         Path to the new model weights
     """
-    raise NotImplementedError()
-
+    # raise NotImplementedError()
+    fragmentation(nodes)
+    process_fragments(nodes)
+