Address embedding test failures

Author: Val Brodsky

libs/labelbox/tests/conftest.py

@@ -1074,20 +1074,15 @@ def configured_project_with_complex_ontology(client, initial_dataset, rand_gen,
     project.delete()
 
 
-@pytest.fixture(scope="session")
+@pytest.fixture
 def embedding(client: Client, environ):
 
     uuid_str = uuid.uuid4().hex
+    time.sleep(randint(1, 5))
     embedding = client.create_embedding(f"sdk-int-{uuid_str}", 8)
     yield embedding
-    # Remove all embeddings on staging
-    if environ == Environ.STAGING:
-        embeddings = client.get_embeddings()
-        for embedding in embeddings:
-            with suppress(LabelboxError):
-                embedding.delete()
-    else:
-        embedding.delete()
+
+    embedding.delete()
 
 
 @pytest.fixture

libs/labelbox/tests/integration/test_embedding.py

@@ -15,22 +15,18 @@ def test_get_embedding_by_id(client: Client, embedding: Embedding):
     e = client.get_embedding_by_id(embedding.id)
     assert e.id == embedding.id
 
-
-def test_get_embedding_by_name(client: Client, embedding: Embedding):
     e = client.get_embedding_by_name(embedding.name)
     assert e.name == embedding.name
 
+    embeddings = client.get_embeddings()
+    assert len(embeddings) > 0
+
 
 def test_get_embedding_by_name_not_found(client: Client):
     with pytest.raises(labelbox.exceptions.ResourceNotFoundError):
         client.get_embedding_by_name("does-not-exist")
 
 
-def test_get_embeddings(client: Client, embedding: Embedding):
-    embeddings = client.get_embeddings()
-    assert len(embeddings) > 0
-
-
 @pytest.mark.parametrize('data_rows', [10], indirect=True)
 def test_import_vectors_from_file(data_rows: List[DataRow],
                                   embedding: Embedding):