Skip to content

Commit ac040ba

Browse files
committed
remember that TF has inconsistent version coverage
1 parent b2e5666 commit ac040ba

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

tests/integ/test_airflow_config.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -515,10 +515,7 @@ def test_sklearn_airflow_config_uploads_data_source_to_s3(
515515

516516
@pytest.mark.canary_quick
517517
def test_tf_airflow_config_uploads_data_source_to_s3(
518-
sagemaker_session,
519-
cpu_instance_type,
520-
tensorflow_training_latest_version,
521-
tensorflow_training_latest_py_version,
518+
sagemaker_session, cpu_instance_type, tf_full_version, tf_full_py_version,
522519
):
523520
with timeout(seconds=AIRFLOW_CONFIG_TIMEOUT_IN_SECONDS):
524521
tf = TensorFlow(
@@ -527,8 +524,8 @@ def test_tf_airflow_config_uploads_data_source_to_s3(
527524
instance_count=SINGLE_INSTANCE_COUNT,
528525
instance_type=cpu_instance_type,
529526
sagemaker_session=sagemaker_session,
530-
framework_version=tensorflow_training_latest_version,
531-
py_version=tensorflow_training_latest_py_version,
527+
framework_version=tf_full_version,
528+
py_version=tf_full_py_version,
532529
metric_definitions=[
533530
{"Name": "train:global_steps", "Regex": r"global_step\/sec:\s(.*)"}
534531
],

0 commit comments

Comments
 (0)