60
60
SINGLE_INSTANCE_COUNT = 1
61
61
62
62
63
- @pytest .mark .canary_quick
64
63
def test_byo_airflow_config_uploads_data_source_to_s3_when_inputs_provided (
65
64
sagemaker_session , cpu_instance_type
66
65
):
@@ -92,7 +91,6 @@ def test_byo_airflow_config_uploads_data_source_to_s3_when_inputs_provided(
92
91
)
93
92
94
93
95
- @pytest .mark .canary_quick
96
94
def test_kmeans_airflow_config_uploads_data_source_to_s3 (sagemaker_session , cpu_instance_type ):
97
95
with timeout (seconds = AIRFLOW_CONFIG_TIMEOUT_IN_SECONDS ):
98
96
kmeans = KMeans (
@@ -153,7 +151,6 @@ def test_fm_airflow_config_uploads_data_source_to_s3(sagemaker_session, cpu_inst
153
151
)
154
152
155
153
156
- @pytest .mark .canary_quick
157
154
def test_ipinsights_airflow_config_uploads_data_source_to_s3 (sagemaker_session , cpu_instance_type ):
158
155
with timeout (seconds = AIRFLOW_CONFIG_TIMEOUT_IN_SECONDS ):
159
156
data_path = os .path .join (DATA_DIR , "ipinsights" )
@@ -214,7 +211,6 @@ def test_knn_airflow_config_uploads_data_source_to_s3(sagemaker_session, cpu_ins
214
211
tests .integ .test_region () in tests .integ .NO_LDA_REGIONS ,
215
212
reason = "LDA image is not supported in certain regions" ,
216
213
)
217
- @pytest .mark .canary_quick
218
214
def test_lda_airflow_config_uploads_data_source_to_s3 (sagemaker_session , cpu_instance_type ):
219
215
with timeout (seconds = AIRFLOW_CONFIG_TIMEOUT_IN_SECONDS ):
220
216
data_path = os .path .join (DATA_DIR , "lda" )
@@ -247,7 +243,6 @@ def test_lda_airflow_config_uploads_data_source_to_s3(sagemaker_session, cpu_ins
247
243
)
248
244
249
245
250
- @pytest .mark .canary_quick
251
246
def test_linearlearner_airflow_config_uploads_data_source_to_s3 (
252
247
sagemaker_session , cpu_instance_type
253
248
):
@@ -312,7 +307,6 @@ def test_linearlearner_airflow_config_uploads_data_source_to_s3(
312
307
)
313
308
314
309
315
- @pytest .mark .canary_quick
316
310
def test_ntm_airflow_config_uploads_data_source_to_s3 (sagemaker_session , cpu_instance_type ):
317
311
with timeout (seconds = AIRFLOW_CONFIG_TIMEOUT_IN_SECONDS ):
318
312
data_path = os .path .join (DATA_DIR , "ntm" )
@@ -346,7 +340,6 @@ def test_ntm_airflow_config_uploads_data_source_to_s3(sagemaker_session, cpu_ins
346
340
)
347
341
348
342
349
- @pytest .mark .canary_quick
350
343
def test_pca_airflow_config_uploads_data_source_to_s3 (sagemaker_session , cpu_instance_type ):
351
344
with timeout (seconds = AIRFLOW_CONFIG_TIMEOUT_IN_SECONDS ):
352
345
pca = PCA (
@@ -373,7 +366,6 @@ def test_pca_airflow_config_uploads_data_source_to_s3(sagemaker_session, cpu_ins
373
366
)
374
367
375
368
376
- @pytest .mark .canary_quick
377
369
def test_rcf_airflow_config_uploads_data_source_to_s3 (sagemaker_session , cpu_instance_type ):
378
370
with timeout (seconds = AIRFLOW_CONFIG_TIMEOUT_IN_SECONDS ):
379
371
# Generate a thousand 14-dimensional datapoints.
@@ -439,7 +431,6 @@ def test_chainer_airflow_config_uploads_data_source_to_s3(
439
431
)
440
432
441
433
442
- @pytest .mark .canary_quick
443
434
def test_mxnet_airflow_config_uploads_data_source_to_s3 (
444
435
sagemaker_session ,
445
436
cpu_instance_type ,
@@ -514,7 +505,6 @@ def test_sklearn_airflow_config_uploads_data_source_to_s3(
514
505
)
515
506
516
507
517
- @pytest .mark .canary_quick
518
508
def test_tf_airflow_config_uploads_data_source_to_s3 (
519
509
sagemaker_session ,
520
510
cpu_instance_type ,
@@ -548,7 +538,6 @@ def test_tf_airflow_config_uploads_data_source_to_s3(
548
538
)
549
539
550
540
551
- @pytest .mark .canary_quick
552
541
def test_xgboost_airflow_config_uploads_data_source_to_s3 (
553
542
sagemaker_session , cpu_instance_type , xgboost_latest_version
554
543
):
@@ -574,7 +563,6 @@ def test_xgboost_airflow_config_uploads_data_source_to_s3(
574
563
)
575
564
576
565
577
- @pytest .mark .canary_quick
578
566
def test_pytorch_airflow_config_uploads_data_source_to_s3_when_inputs_not_provided (
579
567
sagemaker_session ,
580
568
cpu_instance_type ,
0 commit comments