Skip to content

Commit a2fb7a6

Browse files
author
Chuyang Deng
committed
fix import error
1 parent cc927cc commit a2fb7a6

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/integ/test_transformer.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
from sagemaker.mxnet import MXNet
2626
from sagemaker.pytorch import PyTorchModel
2727
from sagemaker.tensorflow import TensorFlow
28-
from sagemaker.tensorflow.defaults import LATEST_SERVING_VERSION
2928
from sagemaker.transformer import Transformer
3029
from sagemaker.estimator import Estimator
3130
from sagemaker.utils import unique_name_from_base
@@ -344,7 +343,9 @@ def test_transform_mxnet_logs(
344343
transformer.wait()
345344

346345

347-
def test_transform_tf_kms_network_isolation(sagemaker_session, cpu_instance_type, tmpdir, tf_full_version, py_version):
346+
def test_transform_tf_kms_network_isolation(
347+
sagemaker_session, cpu_instance_type, tmpdir, tf_full_version, py_version
348+
):
348349
data_path = os.path.join(DATA_DIR, "tensorflow_mnist")
349350

350351
tf = TensorFlow(

0 commit comments

Comments
 (0)