From 6a4844c595c0eb1b5b7f6ca9a69faf6dc3a9bcf5 Mon Sep 17 00:00:00 2001 From: hyunp2 <42776897+hyunp2@users.noreply.github.com> Date: Tue, 17 Oct 2023 23:51:35 -0500 Subject: [PATCH] Update fragment.py --- mofa/fragment.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/mofa/fragment.py b/mofa/fragment.py index 5db92df2..4f1abcd2 100644 --- a/mofa/fragment.py +++ b/mofa/fragment.py @@ -5,4 +5,19 @@ from model import MOFRecord +def fragment_mof_linkers( + starting_model: str | Path, + examples: list[MOFRecord], + num_epochs: int +) -> Path: + """Fragment linkers of MOFs + + Args: + starting_model: Path to the starting weights of the model + examples: Seed examples of linkers (data model TBD) + num_epochs: Number of training epochs + Returns: + Path to the new model weights + """ + raise NotImplementedError()