cleanups

Author: Ubuntu

nucleus/dataset.py

@@ -419,7 +419,6 @@ def items_and_annotations(
             route=f"dataset/{self.id}/exportForTraining",
             requests_command=requests.get,
         )
-        print(api_payload)
         return convert_export_payload(api_payload[EXPORTED_ROWS])
 
     def export_embeddings(

tests/test_dataset.py

@@ -467,8 +467,7 @@ def test_append_and_export(dataset):
             segmentation_annotation,
         ]
     )
-
-    # We don't export everything on the annotations in order to speed up export.
+    # We don't export everything on segmentation annotations in order to speed up export.
     def clear_fields(annotation):
         cleared_annotation = copy.deepcopy(annotation)
         cleared_annotation.metadata = {}
@@ -481,15 +480,11 @@ def sort_labelmap(segmentation_annotation):
 
     exported = dataset.items_and_annotations()
     assert exported[0][ITEM_KEY] == ds_items[0]
-    assert exported[0][ANNOTATIONS_KEY][BOX_TYPE][0] == clear_fields(
-        box_annotation
-    )
+    assert exported[0][ANNOTATIONS_KEY][BOX_TYPE][0] == box_annotation
     assert sort_labelmap(
         exported[0][ANNOTATIONS_KEY][SEGMENTATION_TYPE]
     ) == sort_labelmap(clear_fields(segmentation_annotation))
-    assert exported[0][ANNOTATIONS_KEY][POLYGON_TYPE][0] == clear_fields(
-        polygon_annotation
-    )
+    assert exported[0][ANNOTATIONS_KEY][POLYGON_TYPE][0] == polygon_annotation
 
 
 def test_export_embeddings(CLIENT):

tests/test_indexing.py

@@ -38,6 +38,7 @@ def dataset(CLIENT):
     assert response == {"message": "Beginning dataset deletion..."}
 
 
+@pytest.mark.integration
 def test_index_integration(dataset):
     signed_embeddings_url = TEST_INDEX_EMBEDDINGS_FILE
     create_response = dataset.create_custom_index(