fix: Bump version of sagemaker-training for typing fix (#365)

Author: ajaykarpur

scripts/build_all.py

@@ -54,7 +54,7 @@ def _parse_args():
 # Run docker-login so we can pull the cached image
 login_cmd = subprocess.check_output(
     'aws ecr get-login --no-include-email --registry-id {}'.format(args.account).split())
-print('Executing docker login command: '.format(login_cmd))
+print('Executing docker login command: {}'.format(login_cmd))
 subprocess.check_call(login_cmd.split())
 
 for arch in ['cpu', 'gpu']:

setup.py

@@ -63,7 +63,7 @@ def read_version():
         'Programming Language :: Python :: 3.7',
     ],
 
-    install_requires=['sagemaker-training>=3.5.0', 'numpy', 'scipy', 'sklearn',
+    install_requires=['sagemaker-training>=3.5.1', 'numpy', 'scipy', 'sklearn',
                       'pandas', 'Pillow', 'h5py'],
     extras_require={
         'test': test_dependencies,