diff --git a/deeplabcut/gui/window.py b/deeplabcut/gui/window.py index 0f29cc3577..c91d6050f6 100644 --- a/deeplabcut/gui/window.py +++ b/deeplabcut/gui/window.py @@ -18,6 +18,7 @@ from urllib.error import URLError import warnings import qdarkstyle +from importlib.resources import files import deeplabcut from deeplabcut import auxiliaryfunctions, VERSION, compat @@ -517,10 +518,12 @@ def create_toolbar(self): engine_icon.setStyleSheet("background: transparent;") def _update_icon(engine: str): - pixmap = QPixmap(f"deeplabcut/gui/media/dlc-{engine}.png") - engine_icon.setPixmap( - pixmap.scaled(56, 56, Qt.AspectRatioMode.KeepAspectRatio) - ) + file = files("deeplabcut.gui.media") / f"dlc-{engine}.png" + pixmap = QPixmap(str(file)) + if not pixmap.isNull(): + engine_icon.setPixmap( + pixmap.scaled(56, 56, Qt.AspectRatioMode.KeepAspectRatio) + ) _update_icon("pt" if self.engine == Engine.PYTORCH else "tf")