Skip to content

Commit

Permalink
Refactor Workflow map setter
Browse files Browse the repository at this point in the history
Signed-off-by: liamhuber <[email protected]>
  • Loading branch information
liamhuber committed Jan 17, 2025
1 parent 44d68b2 commit 98213ed
Showing 1 changed file with 7 additions and 9 deletions.
16 changes: 7 additions & 9 deletions pyiron_workflow/workflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -225,10 +225,8 @@ def __init__(
automate_execution: bool = True,
**kwargs,
):
self._inputs_map = None
self._outputs_map = None
self.inputs_map = inputs_map
self.outputs_map = outputs_map
self._inputs_map = self._sanitize_map(inputs_map)
self._outputs_map = self._sanitize_map(outputs_map)
self._inputs = None
self._outputs = None
self.automate_execution: bool = automate_execution
Expand Down Expand Up @@ -269,10 +267,7 @@ def inputs_map(self) -> bidict | None:

@inputs_map.setter
def inputs_map(self, new_map: dict | bidict | None):
self._deduplicate_nones(new_map)
if new_map is not None:
new_map = bidict(new_map)
self._inputs_map = new_map
self._inputs_map = self._sanitize_map(new_map)

@property
def outputs_map(self) -> bidict | None:
Expand All @@ -281,10 +276,13 @@ def outputs_map(self) -> bidict | None:

@outputs_map.setter
def outputs_map(self, new_map: dict | bidict | None):
self._outputs_map = self._sanitize_map(new_map)

def _sanitize_map(self, new_map: dict | bidict | None) -> bidict | None:
self._deduplicate_nones(new_map)
if new_map is not None:
new_map = bidict(new_map)
self._outputs_map = new_map
return new_map

@staticmethod
def _deduplicate_nones(some_map: dict | bidict | None) -> dict | bidict | None:
Expand Down

0 comments on commit 98213ed

Please sign in to comment.