Skip to content

Commit c8963f3

Browse files
authored
use last legacy mode version for --framework-version test arg default (#220)
1 parent b355e85 commit c8963f3

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

test/functional/conftest.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
import boto3
1717
import pytest
1818
from sagemaker import Session
19-
from sagemaker.tensorflow import TensorFlow
2019

2120
logger = logging.getLogger(__name__)
2221
logging.getLogger('boto').setLevel(logging.INFO)
@@ -32,7 +31,7 @@ def pytest_addoption(parser):
3231
parser.addoption('--instance-type')
3332
parser.addoption('--accelerator-type', default=None)
3433
parser.addoption('--region', default='us-west-2')
35-
parser.addoption('--framework-version', default=TensorFlow.LATEST_VERSION)
34+
parser.addoption('--framework-version', default='1.12.0')
3635
parser.addoption('--processor', default='cpu', choices=['gpu', 'cpu'])
3736
parser.addoption('--tag')
3837

0 commit comments

Comments
 (0)