train_test_test: new import

MPoL-dev · Nov 29, 2023 · 35cb36a · 35cb36a
1 parent d74145c
commit 35cb36a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/train_test_test.py b/test/train_test_test.py
@@ -6,7 +6,7 @@
 
 from mpol import losses, precomposed
 from mpol.plot import train_diagnostics_fig
-from mpol.training import TrainTest
+from mpol.training import TrainTest, train_to_dirty_image
 from mpol.constants import *