Update test_fragment.py

globus-labs · Dec 4, 2023 · cadad63 · cadad63
1 parent cf2e686
commit cadad63
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_fragment.py b/tests/test_fragment.py
@@ -4,6 +4,7 @@
 import torch
 mofa.fragment
 from mofa.fragment import fragment_mof_linkers
+from mofa.generator import train_generator, run_generator
 import os
 
 @mark.parametrize('n_atoms', [3])