check if default_args key exist

astronomer · Jan 3, 2025 · ec07050 · ec07050
1 parent 5add203
commit ec07050
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dagfactory/dagfactory.py b/dagfactory/dagfactory.py
@@ -128,7 +128,10 @@ def build_dags(self) -> Dict[str, DAG]:
         default_config: Dict[str, Any] = self.get_default_config()
 
         if global_default_args is not None:
-            default_config = {"default_args": {**global_default_args["default_args"], **default_config["default_args"]}}
+            if "default_args" in default_config and "default_args" in global_default_args:
+                default_config = {
+                    "default_args": {**global_default_args["default_args"], **default_config["default_args"]}
+                }
 
         dags: Dict[str, Any] = {}