fix: propagate log level (#406)

Author: ajaykarpur

setup.py

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