Fix EMR serverless system test (apache#25969)

XD-DENG · Aug 27, 2022 · b75797e · b75797e
1 parent 9e12d48
commit b75797e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/airflow/providers/amazon/aws/sensors/emr.py b/airflow/providers/amazon/aws/sensors/emr.py
@@ -214,8 +214,6 @@ def __init__(
         super().__init__(**kwargs)
 
     def poke(self, context: 'Context') -> bool:
-        state = None
-
         response = self.hook.conn.get_application(applicationId=self.application_id)
 
         state = response['application']['state']

diff --git a/tests/system/providers/amazon/aws/example_emr_serverless.py b/tests/system/providers/amazon/aws/example_emr_serverless.py
@@ -40,6 +40,7 @@
 
 with DAG(
     dag_id=DAG_ID,
+    schedule='@once',
     start_date=datetime(2021, 1, 1),
     tags=['example'],
     catchup=False,
@@ -70,6 +71,7 @@
         release_label='emr-6.6.0',
         job_type="SPARK",
         config={'name': 'new_application'},
+        wait_for_completion=False,
     )
     # [END howto_operator_emr_serverless_create_application]