diff --git a/tests/integ/test_chainer_train.py b/tests/integ/test_chainer_train.py index 0055750c7c..a4afe785c2 100644 --- a/tests/integ/test_chainer_train.py +++ b/tests/integ/test_chainer_train.py @@ -62,6 +62,7 @@ def test_training_with_additional_hyperparameters(sagemaker_session, chainer_ful return chainer.latest_training_job.name +@pytest.mark.continuous_testing def test_attach_deploy(chainer_training_job, sagemaker_session): endpoint_name = 'test-chainer-attach-deploy-{}'.format(sagemaker_timestamp()) diff --git a/tests/integ/test_pytorch_train.py b/tests/integ/test_pytorch_train.py index c7cc312d2c..9c325eb5e9 100644 --- a/tests/integ/test_pytorch_train.py +++ b/tests/integ/test_pytorch_train.py @@ -37,6 +37,7 @@ def fixture_training_job(sagemaker_session, pytorch_full_version): return pytorch.latest_training_job.name +@pytest.mark.continuous_testing def test_sync_fit_deploy(pytorch_training_job, sagemaker_session): # TODO: add tests against local mode when it's ready to be used endpoint_name = 'test-pytorch-sync-fit-attach-deploy{}'.format(sagemaker_timestamp())