refactor: replace ruamel.yaml.main with ruamel.yaml.YAML().load

openedx · Nov 16, 2023 · e323fc3 · e323fc3
1 parent d700f30
commit e323fc3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/edx_repo_tools/codemods/node16/gha_release_workflow_modernizer.py b/edx_repo_tools/codemods/node16/gha_release_workflow_modernizer.py
@@ -7,7 +7,7 @@
 from pathlib import Path
 
 import click
-from ruamel.yaml.main import round_trip_load as yaml_load
+from ruamel.yaml import YAML
 
 from edx_repo_tools.utils import YamlLoader
 
@@ -26,7 +26,8 @@ def _does_nvmrc_exists(self):
 
     def _add_setup_nodejs_env_step(self, step_elements, step_index):
         if self._does_nvmrc_exists():
-            fetch_node_version_step = yaml_load(FETCH_NODE_VERSION_STEP)
+            yaml = YAML()
+            fetch_node_version_step = yaml.load(FETCH_NODE_VERSION_STEP)
             step_elements.insert(
                 step_index, fetch_node_version_step)
         return step_elements