Fix code_widget_class for PW

superstar54 · Nov 21, 2024 · 519179d · 519179d
1 parent 97915df
commit 519179d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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 {}