Fix custom indexing tests (#358)

* Fix model metrics test

* better logs

* Add sleep

Author: shiftsayan

tests/test_indexing.py

@@ -76,6 +76,8 @@ def test_create_custom_index(dataset):
     assert JOB_ID_KEY in job_status_response
     assert MESSAGE_KEY in job_status_response
 
+    job.sleep_until_complete()
+
 
 @pytest.mark.integration
 def test_create_and_delete_custom_index(dataset):
@@ -84,7 +86,8 @@ def test_create_and_delete_custom_index(dataset):
 
     # Starts custom indexing job
     signed_embeddings_url = TEST_INDEX_EMBEDDINGS_FILE
-    dataset.create_custom_index([signed_embeddings_url], embedding_dim=3)
+    job = dataset.create_custom_index([signed_embeddings_url], embedding_dim=3)
+    job.sleep_until_complete()
 
     resp = dataset.set_primary_index(image=True, custom=True)
     assert resp["success"]