Fix support for parent protocol

aiidateam · Sep 26, 2024 · 0778644 · 0778644
1 parent a929450
commit 0778644
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/aiida_wannier90_workflows/workflows/projwfcbands.py b/src/aiida_wannier90_workflows/workflows/projwfcbands.py
@@ -118,15 +118,17 @@ def get_builder_from_protocol(  # pylint: disable=arguments-differ
         # Prepare workchain builder
         builder = cls.get_builder()
 
-        projwfc_overrides = None
-        if overrides:
-            projwfc_overrides = overrides.pop("projwfc", None)
+        protocol_inputs = cls.get_protocol_inputs(
+            protocol=protocol, overrides=overrides
+        )
+
+        projwfc_overrides = protocol_inputs.pop("projwfc", None)
 
         pwbands_builder = PwBandsWorkChain.get_builder_from_protocol(
             code=pw_code,
             structure=structure,
             protocol=protocol,
-            overrides=overrides,
+            overrides=protocol_inputs,
             **kwargs,
         )