Fix deferrable mode execution in WasbPrefixSensor (#31411)

apache · May 23, 2023 · 45b6cfa · 45b6cfa
1 parent df90317
commit 45b6cfa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/airflow/providers/microsoft/azure/triggers/wasb.py b/airflow/providers/microsoft/azure/triggers/wasb.py
@@ -143,7 +143,7 @@ async def run(self) -> AsyncIterator[TriggerEvent]:
         prefix_exists = False
         hook = WasbAsyncHook(wasb_conn_id=self.wasb_conn_id, public_read=self.public_read)
         try:
-            async with hook.blob_service_client:
+            async with await hook.get_async_conn():
                 while not prefix_exists:
                     prefix_exists = await hook.check_for_prefix_async(
                         container_name=self.container_name,