We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 019f9b9 commit 1ec5da1Copy full SHA for 1ec5da1
tests/integ/test_git.py
@@ -15,6 +15,7 @@
15
import os
16
17
import numpy
18
+import pytest
19
import tempfile
20
21
from tests.integ import lock as lock
@@ -30,6 +31,7 @@
30
31
LOCK_PATH = os.path.join(tempfile.gettempdir(), "sagemaker_test_git_lock")
32
33
34
+@pytest.mark.local_mode
35
def test_git_support_with_pytorch(sagemaker_local_session):
36
script_path = "mnist.py"
37
data_path = os.path.join(DATA_DIR, "pytorch_mnist")
@@ -59,6 +61,7 @@ def test_git_support_with_pytorch(sagemaker_local_session):
59
61
predictor.delete_endpoint()
60
62
63
64
65
def test_git_support_with_mxnet(sagemaker_local_session, mxnet_full_version):
66
67
data_path = os.path.join(DATA_DIR, "mxnet_mnist")
0 commit comments