fix: handle missing directive

Author: vberlier

lectern/contrib/relative_location.py

@@ -30,7 +30,7 @@ def __call__(
         fragment: Fragment,
         directives: Mapping[str, Directive],
     ) -> Fragment:
-        if isinstance(directives[fragment.directive], NamespacedResourceDirective):
+        if isinstance(directives.get(fragment.directive), NamespacedResourceDirective):
             name = fragment.expect("name")
             if ":" not in name:
                 fragment = replace(fragment, arguments=[self.ctx.generate.path(name)])

lectern/contrib/yaml_to_json.py

@@ -29,7 +29,7 @@ def beet_default(ctx: Context):
 
 def handle_yaml(fragment: Fragment, directives: Mapping[str, Directive]) -> Fragment:
     """Loader that converts yaml to json."""
-    directive = directives[fragment.directive]
+    directive = directives.get(fragment.directive)
     is_yaml = False
 
     if isinstance(directive, NamespacedResourceDirective):