Use generics to improve type hinting

src/aiidalab_qe/app/configuration/advanced/advanced.py

@@ -14,7 +14,7 @@
 from .smearing import SmearingModel, SmearingSettings
 
 
-class AdvancedSettings(SettingsPanel):
+class AdvancedSettings(SettingsPanel[AdvancedModel]):
     title = "Advanced Settings"
     identifier = "advanced"
 

src/aiidalab_qe/app/configuration/advanced/hubbard/hubbard.py

@@ -6,7 +6,7 @@
 from .model import HubbardModel
 
 
-class HubbardSettings(AdvancedSubSettings):
+class HubbardSettings(AdvancedSubSettings[HubbardModel]):
     identifier = "hubbard"
 
     def __init__(self, model: HubbardModel, **kwargs):

src/aiidalab_qe/app/configuration/advanced/magnetization/magnetization.py

@@ -4,7 +4,7 @@
 from .model import MagnetizationModel
 
 
-class MagnetizationSettings(AdvancedSubSettings):
+class MagnetizationSettings(AdvancedSubSettings[MagnetizationModel]):
     """Widget to set the type of magnetization used in the calculation:
     1) Tot_magnetization: Total majority spin charge - minority spin charge.
     2) Starting magnetization: Starting spin polarization on atomic type 'i' in a spin polarized (LSDA or noncollinear/spin-orbit) calculation.

src/aiidalab_qe/app/configuration/advanced/pseudos/pseudos.py

@@ -19,7 +19,7 @@
 CutoffsPseudoPotentialFamily = GroupFactory("pseudo.family.cutoffs")
 
 
-class PseudoSettings(AdvancedSubSettings):
+class PseudoSettings(AdvancedSubSettings[PseudosModel]):
     identifier = "pseudos"
 
     def __init__(self, model: PseudosModel, **kwargs):

src/aiidalab_qe/app/configuration/advanced/smearing/smearing.py

@@ -4,7 +4,7 @@
 from .model import SmearingModel
 
 
-class SmearingSettings(AdvancedSubSettings):
+class SmearingSettings(AdvancedSubSettings[SmearingModel]):
     identifier = "smearing"
 
     def __init__(self, model: SmearingModel, **kwargs):

src/aiidalab_qe/app/configuration/advanced/subsettings.py

@@ -1,4 +1,5 @@
 import os
+import typing as t
 
 import ipywidgets as ipw
 import traitlets as tl
@@ -26,10 +27,13 @@ def reset(self):
         raise NotImplementedError
 
 
-class AdvancedSubSettings(ipw.VBox):
+M = t.TypeVar("M", bound=AdvancedSubModel)
+
+
+class AdvancedSubSettings(ipw.VBox, t.Generic[M]):
     identifier = "sub"
 
-    def __init__(self, model: AdvancedSubModel, **kwargs):
+    def __init__(self, model: M, **kwargs):
         from aiidalab_qe.common.widgets import LoadingWidget
 
         self.loading_message = LoadingWidget(f"Loading {self.identifier} settings")

src/aiidalab_qe/app/configuration/basic/workflow.py

@@ -5,14 +5,15 @@
 
 import ipywidgets as ipw
 
-from aiidalab_qe.common.panel import SettingsModel, SettingsPanel
+from aiidalab_qe.app.configuration.basic.model import WorkChainModel
+from aiidalab_qe.common.panel import SettingsPanel
 
 
-class WorkChainSettings(SettingsPanel):
+class WorkChainSettings(SettingsPanel[WorkChainModel]):
     title = "Basic Settings"
     identifier = "workchain"
 
-    def __init__(self, model: SettingsModel, **kwargs):
+    def __init__(self, model: WorkChainModel, **kwargs):
         super().__init__(model, **kwargs)
         self._model.observe(
             self._on_input_structure_change,

src/aiidalab_qe/common/panel.py

@@ -8,6 +8,7 @@
 from __future__ import annotations
 
 import os
+import typing as t
 
 import ipywidgets as ipw
 import traitlets as tl
@@ -100,11 +101,14 @@ def reset(self):
         pass
 
 
-class SettingsPanel(Panel):
+M = t.TypeVar("M", bound=SettingsModel)
+
+
+class SettingsPanel(Panel, t.Generic[M]):
     title = "Settings"
     description = ""
 
-    def __init__(self, model: SettingsModel, **kwargs):
+    def __init__(self, model: M, **kwargs):
         from aiidalab_qe.common.widgets import LoadingWidget
 
         self.loading_message = LoadingWidget(f"Loading {self.identifier} settings")

src/aiidalab_qe/plugins/bands/setting.py

@@ -3,9 +3,10 @@
 import ipywidgets as ipw
 
 from aiidalab_qe.common.panel import SettingsPanel
+from aiidalab_qe.plugins.bands.model import BandsModel
 
 
-class Setting(SettingsPanel):
+class Setting(SettingsPanel[BandsModel]):
     title = "Bands Structure"
     identifier = "bands"
 

src/aiidalab_qe/plugins/pdos/setting.py

@@ -9,7 +9,7 @@
 RYDBERG_TO_EV = 13.605703976
 
 
-class Setting(SettingsPanel):
+class Setting(SettingsPanel[PdosModel]):
     title = "PDOS"
     identifier = "pdos"
 

src/aiidalab_qe/plugins/xas/setting.py

@@ -7,7 +7,7 @@
 from .model import XasModel
 
 
-class Setting(SettingsPanel):
+class Setting(SettingsPanel[XasModel]):
     title = "XAS"
     identifier = "xas"
 

src/aiidalab_qe/plugins/xps/setting.py

@@ -7,7 +7,7 @@
 from .model import BASE_URL, XpsModel
 
 
-class Setting(SettingsPanel):
+class Setting(SettingsPanel[XpsModel]):
     title = "XPS"
     identifier = "xps"