fixing duplicated model reference ids

Author: ardila

tests/helpers.py

@@ -7,7 +7,6 @@
 PRESIGN_EXPIRY_SECONDS = 60 * 60 * 24 * 2  # 2 days
 
 TEST_MODEL_NAME = "[PyTest] Test Model"
-TEST_MODEL_REFERENCE = "[PyTest] Test Model Reference" + str(time.time())
 TEST_MODEL_RUN = "[PyTest] Test Model Run"
 TEST_DATASET_NAME = "[PyTest] Test Dataset"
 TEST_SLICE_NAME = "[PyTest] Test Slice"

tests/test_models.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+import time
 import pytest
 from nucleus import (
     Dataset,
@@ -19,7 +20,6 @@
 )
 from helpers import (
     TEST_MODEL_NAME,
-    TEST_MODEL_REFERENCE,
     TEST_MODEL_RUN,
     TEST_PREDS,
 )
@@ -46,8 +46,9 @@ def test_repr(test_object: any):
 def test_model_creation_and_listing(CLIENT, dataset):
     models_before = CLIENT.list_models()
 
+    model_reference = "model_" + time.time()
     # Creation
-    model = CLIENT.add_model(TEST_MODEL_NAME, TEST_MODEL_REFERENCE)
+    model = CLIENT.add_model(TEST_MODEL_NAME, model_reference)
     m_run = model.create_run(TEST_MODEL_RUN, dataset, TEST_PREDS)
     m_run.commit()
 

tests/test_prediction.py

@@ -1,9 +1,8 @@
 import pytest
-
+import time
 from helpers import (
     TEST_DATASET_NAME,
     TEST_MODEL_NAME,
-    TEST_MODEL_REFERENCE,
     TEST_MODEL_RUN,
     TEST_IMG_URLS,
     TEST_BOX_PREDICTIONS,
@@ -58,7 +57,7 @@ def model_run(CLIENT):
     assert ERROR_PAYLOAD not in response.json()
 
     model = CLIENT.add_model(
-        name=TEST_MODEL_NAME, reference_id=TEST_MODEL_REFERENCE
+        name=TEST_MODEL_NAME, reference_id="model_" + str(time.time())
     )
 
     run = model.create_run(name=TEST_MODEL_RUN, dataset=ds, predictions=[])