Skip to content

Commit 4d59c78

Browse files
author
wanyixia
committed
change: revise format
1 parent 8385394 commit 4d59c78

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integ/test_tfs_eia_compilation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def test_compile_and_deploy_with_accelerator(
4646
key_prefix="tensorflow-serving/compiledmodels",
4747
)
4848
bucket = sagemaker_session.default_bucket()
49-
with tests.integ.timeout.timeout_and_delete_endpoint_by_name(endpoint_name, sagemaker_session):
49+
with timeout_and_delete_endpoint_by_name(endpoint_name, sagemaker_session):
5050
model = TensorFlowModel(
5151
model_data=model_data,
5252
role="SageMakerRole",

0 commit comments

Comments
 (0)