Snowflake stage not deleted when load_file() fails

[magdagultekin]

Describe the bug
When loading a file to Snowflake with load_file() operator, the stage in Snowflake is not being deleted if the task fails.

Version

• Astro Runtime: 6.0.3

To Reproduce
Steps to reproduce the behavior:

1. Write the DAG with the following task, make sure there is some incorrect in the CSV file (e.g., one string instead of expected integer):

s3_to_snowflake = aql.load_file(
    task_id="s3_to_snowflake",
    input_file=File(path=f"s3://{S3_BUCKET_NAME}/{S3_FILE_NAME}", filetype=FileType.CSV),
    output_table=Table(
        conn_id=SNOWFLAKE_CONN_ID,
        metadata=Metadata(database=SNOWFLAKE_DATABASE, schema=SNOWFLAKE_SCHEMA),
        name=SNOWFLAKE_TABLE_NAME,
    ),
    if_exists="replace",
)

3. Create Snowflake connection called SNOWFLAKE_CONN_ID
4. Run
5. See not deleted stage in Snowflake

Expected behavior
A stage should be deleted both when the task succeeds and fails.