Skip to content

Commit 203d0f2

Browse files
ardilaUbuntu
authored andcommitted
Refactor a lot of segmentation local upload and async logic (#256)
* work in progress * work in progress * Big refactor to make things cleaner + enable retries properly on infra flakes for local upload * work in progress refactor of annotation upload * Fixed segmentation bugs * Fix one more bug and remove use of annotate_segmentation endpoint * refactor tests and add segmentation local upload test * Tests passing * Review feedback * Initial pass at client changes for prediction segmentation upload * relevant tests pass Co-authored-by: Ubuntu <[email protected]>
1 parent 442eaf8 commit 203d0f2

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

conftest.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,13 @@ def model(CLIENT):
3939
CLIENT.delete_model(model.id)
4040

4141

42+
@pytest.fixture()
43+
def model(CLIENT):
44+
model = CLIENT.create_model(TEST_DATASET_NAME, "fake_reference_id")
45+
yield model
46+
CLIENT.delete_model(model.id)
47+
48+
4249
if __name__ == "__main__":
4350
client = nucleus.NucleusClient(API_KEY)
4451
# ds = client.create_dataset("Test Dataset With Autotags")

0 commit comments

Comments
 (0)