1

scylladb · Jan 12, 2025 · 79f6bfb · 79f6bfb
1 parent d8e6b6d
commit 79f6bfb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/old_tests/test_cmds.py b/old_tests/test_cmds.py
@@ -1,7 +1,7 @@
 import os
 import subprocess
 
-import yaml
+from ruamel.yaml import YAML
 
 from ccmlib import common
 from . import TEST_DIR
@@ -91,6 +91,6 @@ def cluster_create_vnodes_test(self):
         self.validate_output(self.create_cmd(args))
         yaml_path = os.path.join(common.get_default_path(), 'test', 'node1', 'conf', 'cassandra.yaml')
         with open(yaml_path, 'r') as f:
-            data = yaml.safe_load(f)
+            data = YAML().load(f)
 
         self.assertEqual(256, data['num_tokens'])