Skip to content

Commit 0e99263

Browse files
Dan ChoiChoiByungWook
authored andcommitted
change: use sagemaker-training 3.7.0 and enable data parallel integ tests
1 parent b690c43 commit 0e99263

File tree

3 files changed

+1
-4
lines changed

3 files changed

+1
-4
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def read(fname):
5353
'Programming Language :: Python :: 3.7',
5454
],
5555

56-
install_requires=['retrying', 'sagemaker-training>=3.5.1', 'six>=1.12.0'],
56+
install_requires=['retrying', 'sagemaker-training>=3.7.0', 'six>=1.12.0'],
5757
extras_require={
5858
'test': test_dependencies
5959
},

test/integration/local/test_smdataparallel.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,6 @@
2121
from utils.local_mode_utils import assert_files_exist
2222

2323

24-
# TODO: Enable the test once SMDataParallel DLC is publicly accessibly
25-
@pytest.mark.skip(reason="SMDataParallel DLC is not publicly accessible")
2624
@pytest.mark.skip_cpu
2725
@pytest.mark.skip_generic
2826
def test_smdataparallel_training(sagemaker_local_session, image_uri, framework_version, tmpdir):

test/integration/sagemaker/test_smdataparallel.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
from integration.sagemaker.timeout import timeout
2222

2323

24-
@pytest.mark.skip(reason="SMDataParallel DLC is not publicly accessible")
2524
@pytest.mark.skip_cpu
2625
@pytest.mark.skip_generic
2726
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)