From 9b41da72ca2fb5f59e3f06fb228df0f5fa792244 Mon Sep 17 00:00:00 2001
From: Nadia Yakimakha <32335935+nadiaya@users.noreply.github.com>
Date: Thu, 12 Jul 2018 11:22:39 -0700
Subject: [PATCH] Mark pytorch and chainer tests to be used for continuous
 integration.

---
 tests/integ/test_chainer_train.py | 1 +
 tests/integ/test_pytorch_train.py | 1 +
 2 files changed, 2 insertions(+)

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())