Skip to content

Commit b8e7139

Browse files
ardilaUbuntu
andauthored
Fix flaky model fixture (#299)
Co-authored-by: Ubuntu <diego.ardila@scale.com>
1 parent 75e2319 commit b8e7139

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

conftest.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,6 @@
11
import os
2+
import random
3+
import sys
24
from typing import TYPE_CHECKING
35

46
import pytest
@@ -34,7 +36,12 @@ def dataset(CLIENT: "NucleusClient"):
3436

3537
@pytest.fixture()
3638
def model(CLIENT):
37-
model = CLIENT.create_model(TEST_DATASET_NAME, "fake_reference_id")
39+
# Randomly generate an integer between 0 and maximum integer so reference ids
40+
# do not collide during parallel test rusn.
41+
random_postfix = str(random.randint(0, sys.maxsize))
42+
model = CLIENT.create_model(
43+
TEST_DATASET_NAME, "fake_reference_id_" + random_postfix
44+
)
3845
yield model
3946
CLIENT.delete_model(model.id)
4047

0 commit comments

Comments
 (0)