Fix for round_trip_load() YAML error, refs #2219

pull/2217/head
Simon Willison 2023-12-19 10:32:55 -08:00
rodzic 067cc75dfa
commit 89c8ca0f3f
1 zmienionych plików z 3 dodań i 5 usunięć

Wyświetl plik

@ -1,7 +1,7 @@
import json
import textwrap
from yaml import safe_dump
from ruamel.yaml import round_trip_load
from ruamel.yaml import YAML
def metadata_example(cog, data=None, yaml=None):
@ -11,8 +11,7 @@ def metadata_example(cog, data=None, yaml=None):
if yaml:
# dedent it first
yaml = textwrap.dedent(yaml).strip()
# round_trip_load to preserve key order:
data = round_trip_load(yaml)
data = YAML().load(yaml)
output_yaml = yaml
else:
output_yaml = safe_dump(data, sort_keys=False)
@ -27,8 +26,7 @@ def metadata_example(cog, data=None, yaml=None):
def config_example(cog, input):
if type(input) is str:
# round_trip_load to preserve key order:
data = round_trip_load(input)
data = YAML().load(input)
output_yaml = input
else:
data = input