Skip to content

Commit 1ec5da1

Browse files
icywang86ruiknakad
authored andcommitted
fix: add pytest.mark.local_mode annotation to broken tests (aws#884)
1 parent 019f9b9 commit 1ec5da1

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/integ/test_git.py

+3
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
import os
1616

1717
import numpy
18+
import pytest
1819
import tempfile
1920

2021
from tests.integ import lock as lock
@@ -30,6 +31,7 @@
3031
LOCK_PATH = os.path.join(tempfile.gettempdir(), "sagemaker_test_git_lock")
3132

3233

34+
@pytest.mark.local_mode
3335
def test_git_support_with_pytorch(sagemaker_local_session):
3436
script_path = "mnist.py"
3537
data_path = os.path.join(DATA_DIR, "pytorch_mnist")
@@ -59,6 +61,7 @@ def test_git_support_with_pytorch(sagemaker_local_session):
5961
predictor.delete_endpoint()
6062

6163

64+
@pytest.mark.local_mode
6265
def test_git_support_with_mxnet(sagemaker_local_session, mxnet_full_version):
6366
script_path = "mnist.py"
6467
data_path = os.path.join(DATA_DIR, "mxnet_mnist")

0 commit comments

Comments
 (0)