Skip to content

Commit 41483a6

Browse files
laurenyulaurenyu
committed
update integ tests
1 parent d5f80f2 commit 41483a6

File tree

6 files changed

+2
-16
lines changed

6 files changed

+2
-16
lines changed

‎tests/integ/test_airflow_config.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -561,7 +561,6 @@ def test_tf_airflow_config_uploads_data_source_to_s3(sagemaker_session, cpu_inst
561561
train_instance_count=SINGLE_INSTANCE_COUNT,
562562
train_instance_type=cpu_instance_type,
563563
sagemaker_session=sagemaker_session,
564-
script_mode=True,
565564
framework_version=TensorFlow.LATEST_VERSION,
566565
py_version=PYTHON_VERSION,
567566
metric_definitions=[

‎tests/integ/test_horovod.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,6 @@ def test_horovod_local_mode(sagemaker_local_session, instances, processes, tmpdi
5858
train_instance_type="local",
5959
sagemaker_session=sagemaker_local_session,
6060
py_version=integ.PYTHON_VERSION,
61-
script_mode=True,
6261
output_path=output_path,
6362
framework_version="1.12",
6463
distributions={"mpi": {"enabled": True, "processes_per_host": processes}},
@@ -106,7 +105,6 @@ def _create_and_fit_estimator(sagemaker_session, instance_type, tmpdir):
106105
train_instance_type=instance_type,
107106
sagemaker_session=sagemaker_session,
108107
py_version=integ.PYTHON_VERSION,
109-
script_mode=True,
110108
framework_version="1.12",
111109
distributions={"mpi": {"enabled": True}},
112110
)

‎tests/integ/test_tf_script_mode.py renamed to ‎tests/integ/test_tf.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,6 @@ def test_mnist_with_checkpoint_config(
5959
train_instance_count=1,
6060
train_instance_type=instance_type,
6161
sagemaker_session=sagemaker_session,
62-
script_mode=True,
6362
framework_version=tf_full_version,
6463
py_version=py_version,
6564
metric_definitions=[{"Name": "train:global_steps", "Regex": r"global_step\/sec:\s(.*)"}],
@@ -104,7 +103,6 @@ def test_server_side_encryption(sagemaker_session, tf_full_version, py_version):
104103
train_instance_count=1,
105104
train_instance_type="ml.c5.xlarge",
106105
sagemaker_session=sagemaker_session,
107-
script_mode=True,
108106
framework_version=tf_full_version,
109107
py_version=py_version,
110108
code_location=output_path,
@@ -141,7 +139,6 @@ def test_mnist_distributed(sagemaker_session, instance_type, tf_full_version, py
141139
train_instance_type=instance_type,
142140
sagemaker_session=sagemaker_session,
143141
py_version=py_version,
144-
script_mode=True,
145142
framework_version=tf_full_version,
146143
distributions=PARAMETER_SERVER_DISTRIBUTION,
147144
)
@@ -166,7 +163,6 @@ def test_mnist_async(sagemaker_session, cpu_instance_type, tf_full_version, py_v
166163
train_instance_type="ml.c5.4xlarge",
167164
py_version=tests.integ.PYTHON_VERSION,
168165
sagemaker_session=sagemaker_session,
169-
script_mode=True,
170166
# testing py-sdk functionality, no need to run against all TF versions
171167
framework_version=TensorFlow.LATEST_VERSION,
172168
tags=TAGS,
@@ -209,7 +205,6 @@ def test_deploy_with_input_handlers(sagemaker_session, instance_type, tf_full_ve
209205
train_instance_type=instance_type,
210206
py_version=py_version,
211207
sagemaker_session=sagemaker_session,
212-
script_mode=True,
213208
framework_version=tf_full_version,
214209
tags=TAGS,
215210
)

‎tests/integ/test_tf_efs_fsx.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,6 @@ def test_mnist_efs(efs_fsx_setup, sagemaker_session, cpu_instance_type):
6565
train_instance_count=1,
6666
train_instance_type=cpu_instance_type,
6767
sagemaker_session=sagemaker_session,
68-
script_mode=True,
6968
framework_version=TensorFlow.LATEST_VERSION,
7069
py_version=PY_VERSION,
7170
subnets=subnets,
@@ -105,7 +104,6 @@ def test_mnist_lustre(efs_fsx_setup, sagemaker_session, cpu_instance_type):
105104
train_instance_count=1,
106105
train_instance_type=cpu_instance_type,
107106
sagemaker_session=sagemaker_session,
108-
script_mode=True,
109107
framework_version=TensorFlow.LATEST_VERSION,
110108
py_version=PY_VERSION,
111109
subnets=subnets,
@@ -130,7 +128,7 @@ def test_mnist_lustre(efs_fsx_setup, sagemaker_session, cpu_instance_type):
130128
tests.integ.test_region() not in tests.integ.EFS_TEST_ENABLED_REGION,
131129
reason="EFS integration tests need to be fixed before running in all regions.",
132130
)
133-
def test_tuning_tf_script_mode_efs(efs_fsx_setup, sagemaker_session, cpu_instance_type):
131+
def test_tuning_tf_efs(efs_fsx_setup, sagemaker_session, cpu_instance_type):
134132
role = efs_fsx_setup["role_name"]
135133
subnets = [efs_fsx_setup["subnet_id"]]
136134
security_group_ids = efs_fsx_setup["security_group_ids"]
@@ -140,7 +138,6 @@ def test_tuning_tf_script_mode_efs(efs_fsx_setup, sagemaker_session, cpu_instanc
140138
role=role,
141139
train_instance_count=1,
142140
train_instance_type=cpu_instance_type,
143-
script_mode=True,
144141
sagemaker_session=sagemaker_session,
145142
py_version=PY_VERSION,
146143
framework_version=TensorFlow.LATEST_VERSION,
@@ -178,7 +175,7 @@ def test_tuning_tf_script_mode_efs(efs_fsx_setup, sagemaker_session, cpu_instanc
178175
tests.integ.test_region() not in tests.integ.EFS_TEST_ENABLED_REGION,
179176
reason="EFS integration tests need to be fixed before running in all regions.",
180177
)
181-
def test_tuning_tf_script_mode_lustre(efs_fsx_setup, sagemaker_session, cpu_instance_type):
178+
def test_tuning_tf_lustre(efs_fsx_setup, sagemaker_session, cpu_instance_type):
182179
role = efs_fsx_setup["role_name"]
183180
subnets = [efs_fsx_setup["subnet_id"]]
184181
security_group_ids = efs_fsx_setup["security_group_ids"]
@@ -188,7 +185,6 @@ def test_tuning_tf_script_mode_lustre(efs_fsx_setup, sagemaker_session, cpu_inst
188185
role=role,
189186
train_instance_count=1,
190187
train_instance_type=cpu_instance_type,
191-
script_mode=True,
192188
sagemaker_session=sagemaker_session,
193189
py_version=PY_VERSION,
194190
framework_version=TensorFlow.LATEST_VERSION,

‎tests/integ/test_transformer.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -352,7 +352,6 @@ def test_transform_tf_kms_network_isolation(sagemaker_session, cpu_instance_type
352352
train_instance_count=1,
353353
train_instance_type=cpu_instance_type,
354354
framework_version=TensorFlow.LATEST_VERSION,
355-
script_mode=True,
356355
py_version=PYTHON_VERSION,
357356
sagemaker_session=sagemaker_session,
358357
)

‎tests/integ/test_tuner.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -599,7 +599,6 @@ def test_tuning_tf_script_mode(sagemaker_session, cpu_instance_type, tf_full_ver
599599
role="SageMakerRole",
600600
train_instance_count=1,
601601
train_instance_type=cpu_instance_type,
602-
script_mode=True,
603602
sagemaker_session=sagemaker_session,
604603
py_version=PYTHON_VERSION,
605604
framework_version=tf_full_version,

0 commit comments

Comments
 (0)