From 519179d0e3a1223dc31bff9e007b1844d7687d32 Mon Sep 17 00:00:00 2001 From: superstar54 Date: Thu, 21 Nov 2024 11:05:48 +0000 Subject: [PATCH] Fix code_widget_class for PW --- src/aiidalab_qe/app/submission/__init__.py | 1 - src/aiidalab_qe/app/submission/code/model.py | 5 ++++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/aiidalab_qe/app/submission/__init__.py b/src/aiidalab_qe/app/submission/__init__.py index d337d1e7d..885e1d632 100644 --- a/src/aiidalab_qe/app/submission/__init__.py +++ b/src/aiidalab_qe/app/submission/__init__.py @@ -274,7 +274,6 @@ def _set_up_codes(self): "pw": PwCodeModel( description="pw.x", default_calc_job_plugin="quantumespresso.pw", - code_widget_class=PwCodeResourceSetupWidget, ), }, **get_entry_items("aiidalab_qe.properties", "code"), diff --git a/src/aiidalab_qe/app/submission/code/model.py b/src/aiidalab_qe/app/submission/code/model.py index 39ae7d4d8..106cc6f16 100644 --- a/src/aiidalab_qe/app/submission/code/model.py +++ b/src/aiidalab_qe/app/submission/code/model.py @@ -4,7 +4,7 @@ from aiida import orm from aiida.common import NotExistent from aiidalab_qe.common.mvc import Model -from aiidalab_qe.common.widgets import QEAppComputationalResourcesWidget +from aiidalab_qe.common.widgets import QEAppComputationalResourcesWidget, PwCodeResourceSetupWidget class CodeModel(Model): @@ -120,6 +120,9 @@ class PwCodeModel(CodeModel): override = tl.Bool(False) npool = tl.Int(1) + def __init__(self, *, name="", description, default_calc_job_plugin, code_widget_class=PwCodeResourceSetupWidget): + super().__init__(name=name, description=description, default_calc_job_plugin=default_calc_job_plugin, code_widget_class=code_widget_class) + def get_model_state(self) -> dict: parameters = super().get_model_state() parameters["parallelization"] = {"npool": self.npool} if self.override else {}