fix: workaround to print stderr when capture_error is True

Author: ajaykarpur

setup.py

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