Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ class StepFunctionStartExecutionOperator(AwsBaseOperator[StepFunctionHook]):

aws_hook_class = StepFunctionHook
template_fields: Sequence[str] = aws_template_fields(
"state_machine_arn", "name", "input", "is_redrive_execution"
"state_machine_arn", "name", "state_machine_input", "is_redrive_execution"
)
ui_color = "#f9c915"
operator_extra_links = (StateMachineDetailsLink(), StateMachineExecutionsDetailsLink())
Expand All @@ -97,7 +97,7 @@ def __init__(
self.state_machine_arn = state_machine_arn
self.name = name
self.is_redrive_execution = is_redrive_execution
self.input = state_machine_input
self.state_machine_input = state_machine_input
self.waiter_delay = waiter_delay
self.waiter_max_attempts = waiter_max_attempts
self.deferrable = deferrable
Expand All @@ -113,7 +113,7 @@ def execute(self, context: Context):

if not (
execution_arn := self.hook.start_execution(
self.state_machine_arn, self.name, self.input, self.is_redrive_execution
self.state_machine_arn, self.name, self.state_machine_input, self.is_redrive_execution
)
):
raise AirflowException(f"Failed to start State Machine execution for: {self.state_machine_arn}")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -144,7 +144,7 @@ def test_init(self):
assert op.state_machine_arn == STATE_MACHINE_ARN
assert op.state_machine_arn == STATE_MACHINE_ARN
assert op.name == NAME
assert op.input == INPUT
assert op.state_machine_input == INPUT
assert op.hook.aws_conn_id == AWS_CONN_ID
assert op.hook._region_name == REGION_NAME
assert op.hook._verify is False
Expand Down
1 change: 0 additions & 1 deletion scripts/ci/prek/validate_operators_init_exemptions.txt
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,6 @@ providers/amazon/src/airflow/providers/amazon/aws/operators/neptune.py::NeptuneS
providers/amazon/src/airflow/providers/amazon/aws/operators/s3.py::S3DeleteObjectsOperator
providers/amazon/src/airflow/providers/amazon/aws/operators/sagemaker.py::SageMakerCreateNotebookOperator
providers/amazon/src/airflow/providers/amazon/aws/operators/sagemaker.py::SageMakerProcessingOperator
providers/amazon/src/airflow/providers/amazon/aws/operators/step_function.py::StepFunctionStartExecutionOperator
providers/amazon/src/airflow/providers/amazon/aws/transfers/gcs_to_s3.py::GCSToS3Operator
providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/operators/pod.py::KubernetesPodOperator
providers/google/src/airflow/providers/google/cloud/operators/cloud_batch.py::CloudBatchSubmitJobOperator
Expand Down