[imports] Fix ML_LOGGER imports

tum-pbs · Jan 12, 2024 · ae37840 · ae37840
1 parent 7bc85c9
commit ae37840
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 4 deletions.
diff --git a/phi/vis/_dash/colormaps.py b/phi/vis/_dash/colormaps.py
@@ -4,8 +4,6 @@
 
 import numpy as np
 
-from phi.math.backend import ML_LOGGER
-
 ORANGE_WHITE_BLUE = [
     [0.      ,  22.00000005,   1.00000035,  76.0000011 ],
     [0.030334,  28.999875  ,   5.999997  , 114.9999    ],

diff --git a/phi/vis/_dash/dash_gui.py b/phi/vis/_dash/dash_gui.py
@@ -14,7 +14,6 @@
 from .viewer import build_viewers, REFRESH_INTERVAL
 from .player_controls import build_status_bar, build_player_controls, PLAYING, STEP_COMPLETE
 from .._vis_base import Gui, VisModel
-from ...math.backend import ML_LOGGER
 
 
 class DashGui(Gui):

diff --git a/phi/vis/_io.py b/phi/vis/_io.py
@@ -6,7 +6,7 @@
 from phi import math
 from phi.field import Scene
 from phi.math import shape, wrap, channel, spatial, batch
-from phi.math.backend import ML_LOGGER
+from phiml.backend import ML_LOGGER
 
 
 @math.broadcast