update pyyaml version constraint

Author: mollyheamazon

pyproject.toml

@@ -48,7 +48,7 @@ dependencies = [
   "platformdirs",
   "protobuf>=3.12,<6.0",
   "psutil",
-  "PyYAML~=6.0",
+  "PyYAML>=6.0.1",
   "requests",
   "sagemaker-core>=1.0.17,<2.0.0",
   "schema",

src/sagemaker/serve/utils/conda_in_process.yml

@@ -82,7 +82,7 @@ dependencies:
             - python-dateutil>=2.8.2
             - pytz>=2023.3
             - pytz-deprecation-shim>=0.1.0.post0
-            - pyyaml>=5.4.1
+            - pyyaml>=6.0.1
             - regex>=2023.3.23
             - requests>=2.28.2
             - rich>=13.3.4

tox.ini

@@ -86,7 +86,6 @@ commands =
     pip install 'torch==2.0.1+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
     pip install 'torchvision==0.15.2+cpu' -f 'https://download.pytorch.org/whl/torch_stable.html'
     pip install 'dill>=0.3.8'
-    pip install "pyyaml>=6.0.1"
 
     pytest {posargs}
 deps = .[test]