Sfoglia il codice sorgente

Fix import for new surrogate checkpoint

Jérôme BUISINE 4 anni fa
parent
commit
fb86e22cb0
1 ha cambiato i file con 2 aggiunte e 2 eliminazioni
  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):