Skip to content

Commit 8d88ef6

Browse files
committed
fmt
Signed-off-by: Serena Ruan <[email protected]>
1 parent 4e7af7f commit 8d88ef6

File tree

3 files changed

+6
-3
lines changed

3 files changed

+6
-3
lines changed

databricks/sdk/errors/deserializer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,7 @@ def deserialize_error(self, response: requests.Response, response_body: bytes) -
118118
logging.debug("_HtmlErrorParser: no <pre> tag found in error response")
119119
return None
120120

121+
121122
class _ProtobufErrorDeserializer(_ErrorDeserializer):
122123
"""
123124
Parses errors from the Databricks REST API in Protobuf format.

databricks/sdk/errors/parser.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,9 @@
77
from .base import DatabricksError
88
from .customizer import _ErrorCustomizer, _RetryAfterCustomizer
99
from .deserializer import (_EmptyDeserializer, _ErrorDeserializer,
10-
_HtmlErrorDeserializer, _StandardErrorDeserializer,
11-
_StringErrorDeserializer, _ProtobufErrorDeserializer)
10+
_HtmlErrorDeserializer, _ProtobufErrorDeserializer,
11+
_StandardErrorDeserializer,
12+
_StringErrorDeserializer)
1213
from .mapper import _error_mapper
1314
from .private_link import (_get_private_link_validation_error,
1415
_is_private_link_redirect)

tests/test_errors.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,10 @@
55

66
import pytest
77
import requests
8+
from google.rpc import status_pb2
89

910
from databricks.sdk import errors
1011
from databricks.sdk.errors import details
11-
from google.rpc import status_pb2
1212

1313

1414
def fake_response(
@@ -417,6 +417,7 @@ def test_debug_headers_enabled_shows_headers():
417417
assert "X-Databricks-Azure-SP-Management-Token" in error_message
418418
assert "debug-azure-token-67890" in error_message
419419

420+
420421
def test_protobuf_error_deserializer_valid_protobuf():
421422
# Create a valid protobuf Status message
422423
status = status_pb2.Status()

0 commit comments

Comments
 (0)