reorg

Author: srikanth-scale

conftest.py

@@ -55,3 +55,11 @@ def _make_request_patch(
 
     monkeypatch_session.setattr(client, "_make_request", _make_request_patch)
     return client
+
+@pytest.fixture()
+def dataset(CLIENT):
+    ds = CLIENT.create_dataset(TEST_DATASET_NAME)
+    ds.append(TEST_DATASET_ITEMS)
+    yield ds
+
+    CLIENT.delete_dataset(ds.id)

tests/helpers.py

@@ -1,5 +1,4 @@
 from pathlib import Path
-import pytest
 from nucleus import DatasetItem, BoxPrediction
 
 TEST_MODEL_NAME = '[PyTest] Test Model'
@@ -27,14 +26,6 @@
     BoxPrediction('car', 0, 0, 100, 100, '4')
 ]
 
-@pytest.fixture()
-def dataset(CLIENT):
-    ds = CLIENT.create_dataset(TEST_DATASET_NAME)
-    ds.append(TEST_DATASET_ITEMS)
-    yield ds
-
-    CLIENT.delete_dataset(ds.id)
-
 def reference_id_from_url(url):
     return Path(url).name
 

tests/test_models.py

@@ -17,7 +17,6 @@
     DATASET_ID_KEY,
 )
 from helpers import (
-    dataset,
     TEST_MODEL_NAME,
     TEST_MODEL_REFERENCE,
     TEST_MODEL_RUN,