Plot: protect matplotlib.pyplot import form PyQt6 too and remove lingering debug print

This commit is contained in:
Adrian Insaurralde Avalos
2025-12-04 20:37:03 -03:00
committed by Chris Hennes
parent a49d106807
commit fbb65a34fe

View File

@@ -33,13 +33,22 @@ try:
matplotlib.use("QtAgg")
# Force matplotlib to use PySide backend by temporarily unloading PyQt
pyqt5_unloaded = False
pyqt6_unloaded = False
if "PyQt5.QtCore" in sys.modules:
del sys.modules["PyQt5.QtCore"]
import matplotlib.pyplot as plt
pyqt5_unloaded = True
if "PyQt6.QtCore" in sys.modules:
del sys.modules["PyQt6.QtCore"]
pyqt6_unloaded = True
import matplotlib.pyplot as plt
# restore PyQt if unloaded
if pyqt5_unloaded:
import PyQt5.QtCore
else:
print("default matplotlib import")
import matplotlib.pyplot as plt
if pyqt6_unloaded:
import PyQt6.QtCore
from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qtagg import NavigationToolbar2QT as NavigationToolbar