We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b355e85 commit c8963f3Copy full SHA for c8963f3
test/functional/conftest.py
@@ -16,7 +16,6 @@
16
import boto3
17
import pytest
18
from sagemaker import Session
19
-from sagemaker.tensorflow import TensorFlow
20
21
logger = logging.getLogger(__name__)
22
logging.getLogger('boto').setLevel(logging.INFO)
@@ -32,7 +31,7 @@ def pytest_addoption(parser):
32
31
parser.addoption('--instance-type')
33
parser.addoption('--accelerator-type', default=None)
34
parser.addoption('--region', default='us-west-2')
35
- parser.addoption('--framework-version', default=TensorFlow.LATEST_VERSION)
+ parser.addoption('--framework-version', default='1.12.0')
36
parser.addoption('--processor', default='cpu', choices=['gpu', 'cpu'])
37
parser.addoption('--tag')
38
0 commit comments