def arguments(self) -> RequestType:
"""The arguments dict that is used to define the conditional branching in the pipeline."""
return dict(
Conditions=[condition.to_request() for condition in self.conditions],
IfSteps=list_to_request(self.if_steps),
ElseSteps=list_to_request(self.else_steps),
)
def to_request(self) -> RequestType:
"""Gets the request structure for workflow service calls."""
return {
"Version": self._version,
"Metadata": self._metadata,
"Parameters": list_to_request(self.parameters),
"Steps": list_to_request(self.steps),
}
def to_request(self) -> RequestType:
"""Gets the request structure for workflow service calls."""
return {
"Version": self._version,
"Metadata": self._metadata,
"Parameters": list_to_request(self.parameters),
"PipelineExperimentConfig": self.pipeline_experiment_config.to_request()
if self.pipeline_experiment_config is not None
else None,
"Steps": list_to_request(self.steps),
}
def test_step_collection_with_list_to_request():
step_collection = StepCollection(steps=[CustomStep("MyStep1"), CustomStep("MyStep2")])
custom_step = CustomStep("MyStep3")
assert list_to_request([step_collection, custom_step]) == [
{"Name": "MyStep1", "Type": "Training", "Arguments": dict()},
{"Name": "MyStep2", "Type": "Training", "Arguments": dict()},
{"Name": "MyStep3", "Type": "Training", "Arguments": dict()},
]
