Skip to content

Commit c6af5b0

Browse files
authored
test: mark two flaky test (due to race condition) as serial (#5918)
1 parent 7444009 commit c6af5b0

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

sagemaker-train/tests/integ/train/test_custom_scorer_evaluator.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -246,6 +246,7 @@ def test_custom_scorer_evaluator_validation(self):
246246

247247
# @pytest.mark.skip(reason="Built-in metric evaluation - to be enabled when needed")
248248
@pytest.mark.gpu_intensive
249+
@pytest.mark.serial
249250
def test_custom_scorer_with_builtin_metric(self):
250251
"""
251252
Test custom scorer evaluation with built-in metric.
@@ -299,6 +300,7 @@ def test_custom_scorer_with_builtin_metric(self):
299300

300301
# @pytest.mark.skip(reason="Base model only evaluation - not working yet per notebook")
301302
@pytest.mark.gpu_intensive
303+
@pytest.mark.serial
302304
def test_custom_scorer_base_model_only(self):
303305
"""
304306
Test custom scorer evaluation with base model only (no fine-tuned model).

0 commit comments

Comments
 (0)