reformat exception handling

apache · Sep 4, 2024 · 26c6ede · 26c6ede
1 parent f9e015a
commit 26c6ede
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/airflow/providers/microsoft/azure/hooks/asb.py b/airflow/providers/microsoft/azure/hooks/asb.py
@@ -365,8 +365,9 @@ def _process_message(
         else:
             try:
                 message_callback(msg)
-                receiver.complete_message(msg)
             except Exception as e:
                 self.log.error("Error processing message: %s", e)
                 receiver.abandon_message(msg)
                 raise e
+            else:
+                receiver.complete_message(msg)