infra: use sagemaker_session fixture in all Airflow tests (#1312)

laurenyu · web-flow · commit 60ada5bfd4d4 · 2020-02-25T13:51:32.000-08:00
There were a few instances where the fixture was not
passed to the appropriate classes.
diff --git a/tests/integ/test_airflow_config.py b/tests/integ/test_airflow_config.py
@@ -618,6 +618,7 @@ def test_pytorch_airflow_config_uploads_data_source_to_s3_when_inputs_not_provid
             train_instance_count=2,
             train_instance_type=cpu_instance_type,
             hyperparameters={"epochs": 6, "backend": "gloo"},
+            sagemaker_session=sagemaker_session,
         )
 
         training_config = _build_airflow_workflow(
@@ -641,6 +642,7 @@ def test_pytorch_12_airflow_config_uploads_data_source_to_s3_when_inputs_not_pro
             train_instance_count=2,
             train_instance_type=cpu_instance_type,
             hyperparameters={"epochs": 6, "backend": "gloo"},
+            sagemaker_session=sagemaker_session,
         )
 
         training_config = _build_airflow_workflow(

Original file line number	Diff line number	Diff line change
`@@ -618,6 +618,7 @@ def test_pytorch_airflow_config_uploads_data_source_to_s3_when_inputs_not_provid`
`618`	`618`	`train_instance_count=2,`
`619`	`619`	`train_instance_type=cpu_instance_type,`
`620`	`620`	`hyperparameters={"epochs": 6, "backend": "gloo"},`
	`621`	`+ sagemaker_session=sagemaker_session,`
`621`	`622`	`)`
`622`	`623`
`623`	`624`	`training_config = _build_airflow_workflow(`
`@@ -641,6 +642,7 @@ def test_pytorch_12_airflow_config_uploads_data_source_to_s3_when_inputs_not_pro`
`641`	`642`	`train_instance_count=2,`
`642`	`643`	`train_instance_type=cpu_instance_type,`
`643`	`644`	`hyperparameters={"epochs": 6, "backend": "gloo"},`
	`645`	`+ sagemaker_session=sagemaker_session,`
`644`	`646`	`)`
`645`	`647`
`646`	`648`	`training_config = _build_airflow_workflow(`