2727import pyvista as pv
2828from pyvista .plotting .plotter import Plotter as PyVistaPlotter
2929
30- from ansys .tools .visualization_interface import DOCUMENTATION_BUILD , TESTING_MODE
30+ import ansys .tools .visualization_interface as viz_interface
3131from ansys .tools .visualization_interface .types .edge_plot import EdgePlot
3232from ansys .tools .visualization_interface .types .mesh_object_plot import MeshObjectPlot
3333from ansys .tools .visualization_interface .utils .clip_plane import ClipPlane
@@ -74,7 +74,7 @@ def __init__(
7474 # Generate custom scene if ``None`` is provided
7575 if scene is None :
7676 scene = pv .Plotter (plotter_kwargs )
77- if TESTING_MODE :
77+ if viz_interface . TESTING_MODE :
7878 scene .off_screen = True
7979 # If required, use a white background with no gradient
8080 if not color_opts :
@@ -326,7 +326,7 @@ def show(
326326 jupyter_backend = "trame"
327327
328328 # Override Jupyter backend if building docs
329- if DOCUMENTATION_BUILD :
329+ if viz_interface . DOCUMENTATION_BUILD :
330330 jupyter_backend = "html"
331331
332332 # Enabling anti-aliasing by default on scene
@@ -337,7 +337,7 @@ def show(
337337 self .scene .off_screen = True
338338
339339 # If running on testing, set off_screen to True for the plotter
340- if TESTING_MODE or DOCUMENTATION_BUILD :
340+ if viz_interface . TESTING_MODE or viz_interface . DOCUMENTATION_BUILD :
341341 self .scene .off_screen = True
342342
343343 self .scene .show (jupyter_backend = jupyter_backend , ** kwargs )
0 commit comments