add sources only if available (#118)

z3z1ma · Dec 7, 2023 · 784fa82 · 784fa82
1 parent ede5a68
commit 784fa82
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/dbt_osmosis/core/osmosis.py b/src/dbt_osmosis/core/osmosis.py
@@ -686,10 +686,10 @@ def commit_project_restructure_to_disk(
                     target_schema = {"version": 2}
                 elif "version" not in target_schema:
                     target_schema["version"] = 2
-                # Add models and sources to target schema
+                # Add models and sources (if available) to target schema
                 if structure.output["models"]:
                     target_schema.setdefault("models", []).extend(structure.output["models"])
-                if structure.output["sources"]:
+                if structure.output.get("sources") is not None:
                     target_schema.setdefault("sources", []).extend(structure.output["sources"])
                 if not self.dry_run:
                     self.yaml_handler.dump(target_schema, target)