Skip to content

Commit 5250e0d

Browse files
fix: improve yaml decoder performance. (#304)
* fix: improve yaml decoder performance. * fix: improve yaml decoder performance.
1 parent c135fd8 commit 5250e0d

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/tests/test_handler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,7 @@ def test_get_one(self):
211211
assert res.get("info", {}).get("title", "") == "MyChem.info API"
212212

213213
res = self.request("/api/metadata/" + MYGENE_ID + "?format=yaml")
214-
yaml.load(res.text, Loader=yaml.SafeLoader)
214+
yaml.load(res.text, Loader=yaml.CSafeLoader)
215215

216216
def test_get_all(self):
217217
res = self.request("/api/metadata/", method="GET").json()

src/utils/decoder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515

1616
def to_yaml(stream):
1717
try:
18-
data = yaml.load(stream, Loader=yaml.SafeLoader)
18+
data = yaml.load(stream, Loader=yaml.CSafeLoader)
1919
except (yaml.scanner.ScannerError, yaml.parser.ParserError) as err:
2020
raise ValueError(str(err)) from err
2121
if not isinstance(data, dict):

0 commit comments

Comments
 (0)