Skip to content

Commit 1aea0ce

Browse files
committed
[TESTS] Fix vm tests with ruamel update
1 parent ce4f50f commit 1aea0ce

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

‎tests/kt/ktlib/test_vm.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from io import StringIO
33
from unittest.mock import MagicMock, patch
44

5-
import oyaml as yaml
5+
from ruamel.yaml import YAML
66
from pathlib3x import Path
77

88
from kt.ktlib.util import Constants
@@ -144,7 +144,7 @@ def __exit__(self_inner, *args):
144144
content = captured_yaml.get("content", "")
145145
if content.startswith("#cloud-config\n"):
146146
content = content[len("#cloud-config\n") :]
147-
data = yaml.safe_load(content)
147+
data = YAML(typ="safe").load(content)
148148
return data["runcmd"]
149149

150150

0 commit comments

Comments
 (0)