Skip to content

Commit 2d99fce

Browse files
icywang86ruiRui Wang Napieralski
authored andcommitted
fix: mark HuggingFace tests as release tests (aws#2233)
Co-authored-by: Rui Wang Napieralski <[email protected]>
1 parent 73c0232 commit 2d99fce

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/integ/test_huggingface.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,14 @@
1414

1515
import os
1616

17+
import pytest
18+
1719
from sagemaker.huggingface import HuggingFace
1820
from tests.integ import DATA_DIR, TRAINING_DEFAULT_TIMEOUT_MINUTES
1921
from tests.integ.timeout import timeout
2022

2123

24+
@pytest.mark.release
2225
def test_huggingface_training(
2326
sagemaker_session,
2427
gpu_instance_type,

0 commit comments

Comments
 (0)