Skip to content

Commit 94d271e

Browse files
committed
Merge branch 'refactor-tensorflow-general' of https://github.com/eslesar-aws/sagemaker-python-sdk into refactor-tensorflow-general
2 parents 122f066 + 98ed2a6 commit 94d271e

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

doc/overview.rst

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ SageMaker Python SDK provides several high-level abstractions for working with A
99
- **Predictors**: Provide real-time inference and transformation using Python data-types against a SageMaker endpoint.
1010
- **Session**: Provides a collection of methods for working with SageMaker resources.
1111

12-
``Estimator`` and ``Model`` implementations for MXNet, TensorFlow, Chainer, PyTorch, scikit-Learn, Amazon SageMaker built-in algorithms, Reinforcement Learning, are included.
12+
``Estimator`` and ``Model`` implementations for MXNet, TensorFlow, Chainer, PyTorch, scikit-learn, Amazon SageMaker built-in algorithms, Reinforcement Learning, are included.
1313
There's also an ``Estimator`` that runs SageMaker compatible custom Docker containers, enabling you to run your own ML algorithms by using the SageMaker Python SDK.
1414

1515
.. contents::
@@ -30,7 +30,7 @@ After you train a model, you can save it, and then serve the model as an endpoin
3030
Prepare a Training script
3131
=========================
3232

33-
Your training script must be a Python 2.7 or 3.5 compatible source file.
33+
Your training script must be a Python 2.7 or 3.6 compatible source file.
3434

3535
The training script is very similar to a training script you might run outside of SageMaker, but you can access useful properties about the training environment through various environment variables, including the following:
3636

@@ -984,4 +984,4 @@ Example code using the TensorFlow predictor:
984984
from sagemaker.tensorflow import TensorFlowPredictor
985985
986986
predictor = TensorFlowPredictor('myexistingendpoint')
987-
result = predictor.predict(['my request body'])
987+
result = predictor.predict(['my request body'])

tests/integ/test_git.py

Lines changed: 3 additions & 0 deletions
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)