Skip to content

Commit 06d27d3

Browse files
committed
Fix device combobox when no GPUs are available
1 parent a606e2f commit 06d27d3

File tree

1 file changed

+16
-18
lines changed

1 file changed

+16
-18
lines changed

ui_function.py

Lines changed: 16 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1252,26 +1252,24 @@ def init_run(main_window):
12521252
main_window : QMainWindow
12531253
Main window of the application.
12541254
"""
1255+
# Create GPU input field
1256+
# main_window.ui.device_input = CheckableComboBox(main_window.ui.gpu_list_frame)
1257+
main_window.ui.device_input = QComboBox(main_window.ui.gpu_list_frame)
1258+
main_window.ui.device_input.setObjectName(u"device_input")
1259+
main_window.ui.device_input.setMinimumSize(QSize(400, 30))
1260+
main_window.ui.device_input.setMaximumSize(QSize(400, 30))
1261+
font = QFont()
1262+
font.setFamily(u"DejaVu Math TeX Gyre")
1263+
font.setPointSize(12)
1264+
main_window.ui.device_input.setFont(font)
1265+
main_window.ui.verticalLayout_40.addWidget(main_window.ui.device_input, 0, Qt.AlignHCenter)
1266+
main_window.ui.device_input.addItem("CPU")
1267+
12551268
if len(main_window.cfg.settings['GPUs']) > 0:
1256-
# Create GPU input field
1257-
# main_window.ui.gpu_input = CheckableComboBox(main_window.ui.gpu_list_frame)
1258-
main_window.ui.gpu_input = QComboBox(main_window.ui.gpu_list_frame)
1259-
main_window.ui.gpu_input.setObjectName(u"gpu_input")
1260-
main_window.ui.gpu_input.setMinimumSize(QSize(400, 30))
1261-
main_window.ui.gpu_input.setMaximumSize(QSize(400, 30))
1262-
font = QFont()
1263-
font.setFamily(u"DejaVu Math TeX Gyre")
1264-
font.setPointSize(12)
1265-
main_window.ui.gpu_input.setFont(font)
1266-
main_window.ui.verticalLayout_40.addWidget(main_window.ui.gpu_input, 0, Qt.AlignHCenter)
12671269
# for i, gpu in enumerate(main_window.cfg.settings['GPUs']):
1268-
# main_window.ui.gpu_input.addItem("{} : {} ".format(i, gpu.name), check=True if i == 0 else False)
1269-
main_window.ui.gpu_input.addItem("CPU")
1270+
# main_window.ui.device_input.addItem("{} : {} ".format(i, gpu.name), check=True if i == 0 else False)
12701271
for i, gpu in enumerate(main_window.cfg.settings['GPUs']):
1271-
main_window.ui.gpu_input.addItem("GPU {} : {} ".format(i, gpu.name))
1272-
else:
1273-
main_window.ui.gpu_list_frame.setVisible(False)
1274-
main_window.ui.run_workflow_frame.setFrameShape(QFrame.NoFrame)
1272+
main_window.ui.device_input.addItem("GPU {} : {} ".format(i, gpu.name))
12751273

12761274
def build_container(main_window):
12771275
"""
@@ -1361,7 +1359,7 @@ def run_biapy(main_window):
13611359
return
13621360

13631361
# Device selection
1364-
device = get_text(main_window.ui.gpu_input)
1362+
device = get_text(main_window.ui.device_input)
13651363
use_gpu = True if "GPU" in device else False
13661364

13671365
# Check if a pull is necessary

0 commit comments

Comments
 (0)