Update test_generator.py

globus-labs · Nov 8, 2023 · cc6c2ee · cc6c2ee
1 parent 80e4f79
commit cc6c2ee
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_generator.py b/tests/test_generator.py
@@ -3,7 +3,8 @@
 from mofa.generator import train_generator, run_generator
 import numpy as np
 import torch
-from 
+from mofa.utils.src.lightning import DDPM
+
 
 @fixture()
 def load_model():