Added callback_file & callback_name to default_args DAG level and tes…

…ts: fmt-check
astronomer · Sep 20, 2024 · 5f44816 · 5f44816
1 parent 173f69a
commit 5f44816
Showing 1 changed file with 14 additions and 8 deletions.
diff --git a/dagfactory/dagbuilder.py b/dagfactory/dagbuilder.py
@@ -257,21 +257,27 @@ def get_dag_params(self) -> Dict[str, Any]:
         if utils.check_dict_key(
             dag_params["default_args"], "on_success_callback_name"
         ) and utils.check_dict_key(
-            dag_params["default_args"], "on_success_callback_file"):
+            dag_params["default_args"], "on_success_callback_file"
+        ):
 
-            dag_params["default_args"]["on_success_callback"]: Callable = utils.get_python_callable(
-                dag_params["default_args"]["on_success_callback_name"],
-                dag_params["default_args"]["on_success_callback_file"],
+            dag_params["default_args"]["on_success_callback"]: Callable = (
+                utils.get_python_callable(
+                    dag_params["default_args"]["on_success_callback_name"],
+                    dag_params["default_args"]["on_success_callback_file"],
+                )
             )
 
         if utils.check_dict_key(
             dag_params["default_args"], "on_failure_callback_name"
         ) and utils.check_dict_key(
-            dag_params["default_args"], "on_failure_callback_file"):
+            dag_params["default_args"], "on_failure_callback_file"
+        ):
 
-            dag_params["default_args"]["on_failure_callback"]: Callable = utils.get_python_callable(
-                dag_params["default_args"]["on_failure_callback_name"],
-                dag_params["default_args"]["on_failure_callback_file"],
+            dag_params["default_args"]["on_failure_callback"]: Callable = (
+                utils.get_python_callable(
+                    dag_params["default_args"]["on_failure_callback_name"],
+                    dag_params["default_args"]["on_failure_callback_file"],
+                )
             )
 
         if utils.check_dict_key(dag_params, "template_searchpath"):