Skip to content

Context is not preserved after execution returns from deferred state #39503

Description

@mxuan0

Apache Airflow version

Other Airflow 2 version (please specify below)

If "Other Airflow 2 version" selected, which one?

2.7.2

What happened?

We are using 8.7.1. version of airflow provider package and set deferrable to True when using SageMakerTransformOperator. We set define the model name with something like "project-dag_name-date-uuid", where uuid is to deduplicate model names upon retry. However, when the execution gets back from deferred state, the original uuid is not preserved and there would be a new uuid so the operator cannot find a model with the new model name and it would fail.

What you think should happen instead?

The original UUID should be preserved

How to reproduce

In the operator config, set "ModelName": f"{name}-{str(uuid4())[:8]}". The run the DAG.

Operating System

amazon Linux AMI

Versions of Apache Airflow Providers

8.7.1

Deployment

Amazon (AWS) MWAA

Deployment details

No response

Anything else?

No response

Are you willing to submit PR?

  • Yes I am willing to submit a PR!

Code of Conduct

Activity

  1. potiuk commented on May 9, 2024

    @potiuk
    Member

    Can you check if it's the same in latest provider please @mxuan0 ?

  2. vincbeck commented on May 16, 2024

    @vincbeck
    Contributor

    Thanks @mxuan0 for posting the issue, it is indeed a bug, here is a fix: #39671

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions