fix: upgrade to sagemaker-training 3.7.1 (#414)

Co-authored-by: Rui Wang Napieralski <wru@amazon.com>

Author: icywang86rui

setup.py

@@ -72,7 +72,7 @@ def read_version():
         "Programming Language :: Python :: 3.7",
     ],
     install_requires=[
-        "sagemaker-training>=3.7.0",
+        "sagemaker-training>=3.7.1",
         "numpy",
         "scipy",
         "sklearn",