Remove yaml for config file

Author: Yongyan Rao

src/sagemaker/fw_utils.py

@@ -22,7 +22,6 @@
 import tempfile
 from collections import namedtuple
 from typing import Optional, Union, Dict
-import yaml
 
 import sagemaker.image_uris
 from sagemaker.session_settings import SessionSettings
@@ -222,7 +221,7 @@ def parse_mp_parameters(params):
 
     Raises:
         ValueError: if params is not a string or a dict, or
-                    the config file cannot be parsed as json or yaml.
+                    the config file cannot be parsed as json.
     """
     parsed = None
     if isinstance(params, dict):
@@ -232,19 +231,15 @@ def parse_mp_parameters(params):
             with open(params, "r") as fp:
                 parsed = json.load(fp)
         except json.decoder.JSONDecodeError:
-            try:
-                with open(params, "r") as fp:
-                    parsed = yaml.load(fp)
-            except yaml.YAMLError:
-                pass
+            pass
     else:
         raise ValueError(
             f"Expected a string path to an existing modelparallel config, or a dictionary. "
             f"Received: {params}."
         )
 
     if parsed is None:
-        raise ValueError(f"Cannot parse {params} as a json or yaml file.")
+        raise ValueError(f"Cannot parse {params} as a json file.")
 
     return parsed
 

tests/unit/test_fw_utils.py

@@ -18,7 +18,6 @@
 import tarfile
 from contextlib import contextmanager
 from itertools import product
-import yaml
 
 import pytest
 
@@ -226,24 +225,6 @@ def test_parse_mp_parameters_input_str_json():
     os.remove(json_file_path)
 
 
-def test_parse_mp_parameters_input_str_yaml():
-    mp_parameters = {
-        "partitions": 1,
-        "tensor_parallel_degree": 2,
-        "microbatches": 1,
-        "optimize": "speed",
-        "pipeline": "interleaved",
-        "ddp": 1,
-        "auto_partition": False,
-        "default_partition": 0,
-    }
-    yaml_file_path = "./params.yaml"
-    with open(yaml_file_path, "x") as fp:
-        yaml.dump(mp_parameters, fp)
-    assert mp_parameters == fw_utils.parse_mp_parameters(yaml_file_path)
-    os.remove(yaml_file_path)
-
-
 def test_parse_mp_parameters_input_not_exit():
     with pytest.raises(ValueError):
         fw_utils.parse_mp_parameters(" !@#$%^&*()path probably in not there.!@#$%^&*()")