fixed lint

Author: daniel-sanche

google/cloud/bigtable/data/_metrics/data_model.py

@@ -27,7 +27,9 @@
 from dataclasses import field
 from grpc import StatusCode
 
-import google.cloud.bigtable.data.exceptions as bt_exceptions
+from google.cloud.bigtable.data.exceptions import MutationsExceptionGroup
+from google.cloud.bigtable.data.exceptions import ShardedReadRowsExceptionGroup
+from google.cloud.bigtable.data.exceptions import RetryExceptionGroup
 from google.cloud.bigtable_v2.types.response_params import ResponseParams
 from google.protobuf.message import DecodeError
 
@@ -411,10 +413,13 @@ def _exc_to_status(exc: Exception) -> StatusCode:
           - exc: The exception to extract the status code from.
         """
         # parse bigtable custom exceptions
-        if isinstance(exc, (bt_exceptions.MutationsExceptionGroup, bt_exceptions.ShardedReadRowsExceptionGroup)) and exc.exceptions:
+        if (
+            isinstance(exc, (MutationsExceptionGroup, ShardedReadRowsExceptionGroup))
+            and exc.exceptions
+        ):
             # grab the RetryExceptionGroup from the most last failed mutation/shard
             exc = exc.exceptions[-1].__cause__
-        if isinstance(exc, bt_exceptions.RetryExceptionGroup):
+        if isinstance(exc, RetryExceptionGroup):
             # grab the last exception in the retry group
             exc = exc.exceptions[-1]
         # parse grpc exceptions

google/cloud/bigtable/data/_metrics/handlers/opentelemetry.py

@@ -212,7 +212,8 @@ def on_attempt_complete(
             combined_throttling += op.flow_throttling_time_ms
         self.otel.throttling_latencies.record(combined_throttling, labels)
         self.otel.application_latencies.record(
-            attempt.application_blocking_time_ms + attempt.backoff_before_attempt_ms, labels
+            attempt.application_blocking_time_ms + attempt.backoff_before_attempt_ms,
+            labels,
         )
         if (
             op.op_type == OperationType.READ_ROWS
@@ -229,6 +230,4 @@ def on_attempt_complete(
         else:
             # gfe headers not attached. Record a connectivity error.
             # TODO: this should not be recorded as an error when direct path is enabled
-            self.otel.connectivity_error_count.add(
-                1, {"status": status, **labels}
-            )
+            self.otel.connectivity_error_count.add(1, {"status": status, **labels})