From 0832a62c7a2c963712521a2cdb8b19b5af48d5a4 Mon Sep 17 00:00:00 2001 From: Rui Wang Date: Wed, 26 Jun 2019 09:47:38 -0700 Subject: [PATCH] fix: add pytest.mark.local_mode annotation to broken tests --- tests/integ/test_git.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/integ/test_git.py b/tests/integ/test_git.py index 0f01455fe9..7d5c1b76ae 100644 --- a/tests/integ/test_git.py +++ b/tests/integ/test_git.py @@ -15,6 +15,7 @@ import os import numpy +import pytest import tempfile from tests.integ import lock as lock @@ -30,6 +31,7 @@ LOCK_PATH = os.path.join(tempfile.gettempdir(), "sagemaker_test_git_lock") +@pytest.mark.local_mode def test_git_support_with_pytorch(sagemaker_local_session): script_path = "mnist.py" data_path = os.path.join(DATA_DIR, "pytorch_mnist") @@ -59,6 +61,7 @@ def test_git_support_with_pytorch(sagemaker_local_session): predictor.delete_endpoint() +@pytest.mark.local_mode def test_git_support_with_mxnet(sagemaker_local_session, mxnet_full_version): script_path = "mnist.py" data_path = os.path.join(DATA_DIR, "mxnet_mnist")