diff --git a/dagger/utilities/module.py b/dagger/utilities/module.py
index 7f33690..2b48bfe 100644
--- a/dagger/utilities/module.py
+++ b/dagger/utilities/module.py
@@ -27,7 +27,7 @@ def __init__(self, path_to_config, target_dir, jinja_parameters=None):
                 f"{path.join(self._directory, task)}.yaml"
             )
 
-        self._branches_to_generate = config["branches_to_generate"]
+        self._branches_to_generate = config.get("branches_to_generate", {})
         self._override_parameters = config.get("override_parameters", {})
         self._default_parameters = config.get("default_parameters", {})
         self._jinja_parameters = jinja_parameters or {}
@@ -96,13 +96,17 @@ def dump_yaml(yaml_str, yaml_path):
             )
 
     def generate_task_configs(self):
+        if not self._branches_to_generate:
+            _logger.warning("No branches to generate found in config")
+            return
+
         for branch_name, attrs in self._branches_to_generate.items():
             attrs = {} if attrs is None else attrs
             _logger.info(f"Generating tasks for branch {branch_name}")
             template_parameters = {}
             template_parameters.update(self._default_parameters or {})
             template_parameters.update(attrs)
-            template_parameters['branch_name'] = branch_name
+            template_parameters["branch_name"] = branch_name
             template_parameters.update(self._jinja_parameters)
 
             for task, task_yaml in self._tasks.items():