Skip to content

Commit 5681b95

Browse files
vbrodskyVal Brodsky
authored andcommitted
Fix exception type for labeling service test (#1835)
1 parent a81e3a6 commit 5681b95

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

libs/labelbox/src/labelbox/client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,10 @@
77
import time
88
import urllib.parse
99
from collections import defaultdict
10+
from datetime import datetime, timezone
1011
from types import MappingProxyType
1112
from typing import Any, Callable, Dict, List, Optional, Set, Union, overload
1213

13-
import lbox.exceptions
1414
import requests
1515
import requests.exceptions
1616
from google.api_core import retry

libs/labelbox/tests/integration/test_labeling_service.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import pytest
2-
from lbox.exceptions import LabelboxError, ResourceNotFoundError
2+
from lbox.exceptions import MalformedQueryException, ResourceNotFoundError
33

44
from labelbox.schema.labeling_service import LabelingServiceStatus
55

@@ -51,7 +51,7 @@ def test_request_labeling_service_moe_project(
5151

5252
labeling_service = project.get_labeling_service()
5353
with pytest.raises(
54-
LabelboxError,
54+
MalformedQueryException,
5555
match='[{"errorType":"PROJECT_MODEL_CONFIG","errorMessage":"Project model config is not completed"}]',
5656
):
5757
labeling_service.request()
@@ -73,5 +73,5 @@ def test_request_labeling_service_incomplete_requirements(ontology, project):
7373
): # No labeling service by default
7474
labeling_service.request()
7575
project.connect_ontology(ontology)
76-
with pytest.raises(LabelboxError):
76+
with pytest.raises(MalformedQueryException):
7777
labeling_service.request()

0 commit comments

Comments
 (0)