Allow empty defaults.yaml and groups.yaml (#7)

nornir-automation · Jun 22, 2020 · fd9e8e8 · fd9e8e8
1 parent d911461
commit fd9e8e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nornir_utils/plugins/inventory/yaml_inventory.py b/nornir_utils/plugins/inventory/yaml_inventory.py
@@ -93,7 +93,7 @@ def load(self) -> Inventory:
 
         if self.defaults_file.exists():
             with open(self.defaults_file, "r") as f:
-                defaults_dict = yml.load(f)
+                defaults_dict = yml.load(f) or {}
             defaults = _get_defaults(defaults_dict)
         else:
             defaults = Defaults()
@@ -108,7 +108,7 @@ def load(self) -> Inventory:
         groups = Groups()
         if self.group_file.exists():
             with open(self.group_file, "r") as f:
-                groups_dict = yml.load(f)
+                groups_dict = yml.load(f) or {}
 
             for n, g in groups_dict.items():
                 groups[n] = _get_inventory_element(Group, g, n, defaults)