Parcourir la source

Fix import for new surrogate checkpoint

Jérôme BUISINE il y a 3 ans
Parent
commit
fb86e22cb0
1 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. 2 2
      optimization/callbacks/SurrogateMonoCheckpoint.py

+ 2 - 2
optimization/callbacks/SurrogateMonoCheckpoint.py

@@ -7,8 +7,8 @@ import logging
 import numpy as np
 
 # module imports
-from macop.callbacks.Callback import Callback
-from macop.utils.color import macop_text, macop_line
+from macop.callbacks.base.Callback import Callback
+from macop.utils.progress import macop_text, macop_line
 
 
 class SurrogateMonoCheckpoint(Callback):