Skip to content

Commit 1ec470f

Browse files
authored
Mark pytorch and chainer tests to be used for continuous integration. (#290)
1 parent a667801 commit 1ec470f

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

tests/integ/test_chainer_train.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ def test_training_with_additional_hyperparameters(sagemaker_session, chainer_ful
6262
return chainer.latest_training_job.name
6363

6464

65+
@pytest.mark.continuous_testing
6566
def test_attach_deploy(chainer_training_job, sagemaker_session):
6667
endpoint_name = 'test-chainer-attach-deploy-{}'.format(sagemaker_timestamp())
6768

tests/integ/test_pytorch_train.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ def fixture_training_job(sagemaker_session, pytorch_full_version):
3737
return pytorch.latest_training_job.name
3838

3939

40+
@pytest.mark.continuous_testing
4041
def test_sync_fit_deploy(pytorch_training_job, sagemaker_session):
4142
# TODO: add tests against local mode when it's ready to be used
4243
endpoint_name = 'test-pytorch-sync-fit-attach-deploy{}'.format(sagemaker_timestamp())

0 commit comments

Comments
 (0)