Fix tests

Author: Val Brodsky

libs/labelbox/src/labelbox/client.py

@@ -202,9 +202,7 @@ def upload_file(self, path: str) -> str:
             )
 
     @retry.Retry(
-        predicate=retry.if_exception_type(
-            lbox.exceptions.InternalServerError
-        )
+        predicate=retry.if_exception_type(lbox.exceptions.InternalServerError)
     )
     def upload_data(
         self,
@@ -310,9 +308,7 @@ def _get_single(self, db_object_type, uid):
         res = self.execute(query_str, params)
         res = res and res.get(utils.camel_case(db_object_type.type_name()))
         if res is None:
-            raise lbox.exceptions.ResourceNotFoundError(
-                db_object_type, params
-            )
+            raise lbox.exceptions.ResourceNotFoundError(db_object_type, params)
         else:
             return db_object_type(self, res)
 
@@ -1975,9 +1971,7 @@ def is_feature_schema_archived(
                 )
 
         elif response.status_code == 404:
-            raise lbox.exceptions.ResourceNotFoundError(
-                Ontology, ontology_id
-            )
+            raise lbox.exceptions.ResourceNotFoundError(Ontology, ontology_id)
         else:
             raise lbox.exceptions.LabelboxError(
                 "Failed to get the feature schema archived status."
@@ -2006,9 +2000,7 @@ def get_model_slice(self, slice_id) -> ModelSlice:
         """
         res = self.execute(query_str, {"id": slice_id})
         if res is None or res["getSavedQuery"] is None:
-            raise lbox.exceptions.ResourceNotFoundError(
-                ModelSlice, slice_id
-            )
+            raise lbox.exceptions.ResourceNotFoundError(ModelSlice, slice_id)
 
         return Entity.ModelSlice(self, res["getSavedQuery"])
 
@@ -2316,9 +2308,7 @@ def get_embedding_by_name(self, name: str) -> Embedding:
         for e in embeddings:
             if e.name == name:
                 return e
-        raise lbox.exceptions.ResourceNotFoundError(
-            Embedding, dict(name=name)
-        )
+        raise lbox.exceptions.ResourceNotFoundError(Embedding, dict(name=name))
 
     def upsert_label_feedback(
         self, label_id: str, feedback: str, scores: Dict[str, float]
@@ -2365,8 +2355,7 @@ def upsert_label_feedback(
         scores_raw = res["upsertAutoQaLabelFeedback"]["scores"]
 
         return [
-            labelbox.LabelScore(name=x["name"], score=x["score"])
-            for x in scores_raw
+            LabelScore(name=x["name"], score=x["score"]) for x in scores_raw
         ]
 
     def get_labeling_service_dashboards(

libs/labelbox/tests/integration/test_client_errors.py

@@ -73,7 +73,7 @@ def test_resource_not_found_error(client):
 
 def test_network_error(client):
     client = labelbox.client.Client(
-        api_key=client.api_key, endpoint="not_a_valid_URL"
+        api_key=client._request_client.api_key, endpoint="not_a_valid_URL"
     )
 
     with pytest.raises(lbox.exceptions.NetworkError) as excinfo:

libs/labelbox/tests/integration/test_foundry.py

@@ -1,7 +1,8 @@
-import labelbox as lb
 import pytest
-from labelbox.schema.foundry.app import App
+from lbox.exceptions import LabelboxError, ResourceNotFoundError
 
+import labelbox as lb
+from labelbox.schema.foundry.app import App
 from labelbox.schema.foundry.foundry_client import FoundryClient
 
 # Yolo object detection model id
@@ -97,7 +98,7 @@ def test_get_app(foundry_client, app):
 
 
 def test_get_app_with_invalid_id(foundry_client):
-    with pytest.raises(lb.exceptions.ResourceNotFoundError):
+    with pytest.raises(ResourceNotFoundError):
         foundry_client._get_app("invalid-id")
 
 
@@ -144,7 +145,7 @@ def test_run_foundry_app_returns_model_run_id(
 def test_run_foundry_with_invalid_data_row_id(foundry_client, app, random_str):
     invalid_datarow_id = "invalid-global-key"
     data_rows = lb.GlobalKeys([invalid_datarow_id])
-    with pytest.raises(lb.exceptions.LabelboxError) as exception:
+    with pytest.raises(LabelboxError) as exception:
         foundry_client.run_app(
             model_run_name=f"test-app-with-invalid-datarow-id-{random_str}",
             data_rows=data_rows,
@@ -156,7 +157,7 @@ def test_run_foundry_with_invalid_data_row_id(foundry_client, app, random_str):
 def test_run_foundry_with_invalid_global_key(foundry_client, app, random_str):
     invalid_global_key = "invalid-global-key"
     data_rows = lb.GlobalKeys([invalid_global_key])
-    with pytest.raises(lb.exceptions.LabelboxError) as exception:
+    with pytest.raises(LabelboxError) as exception:
         foundry_client.run_app(
             model_run_name=f"test-app-with-invalid-global-key-{random_str}",
             data_rows=data_rows,