fixed lint

Author: daniel-sanche

google/cloud/bigtable/data/_async/_mutate_rows.py

@@ -16,8 +16,6 @@
 
 from typing import Sequence, TYPE_CHECKING
 
-from functools import partial
-
 from google.api_core import exceptions as core_exceptions
 from google.api_core import retry as retries
 import google.cloud.bigtable_v2.types.bigtable as types_pb

google/cloud/bigtable/data/_async/_read_rows.py

@@ -18,7 +18,6 @@
 from typing import Sequence, TYPE_CHECKING
 
 import time
-from functools import partial
 
 from google.cloud.bigtable_v2.types import ReadRowsRequest as ReadRowsRequestPB
 from google.cloud.bigtable_v2.types import ReadRowsResponse as ReadRowsResponsePB

google/cloud/bigtable/data/_async/metrics_interceptor.py

@@ -156,7 +156,10 @@ async def intercept_unary_stream(
     ):
         async def response_wrapper(call):
             # only track has_first response for READ_ROWS
-            has_first_response = operation.first_response_latency_ns is not None or operation.op_type != OperationType.READ_ROWS
+            has_first_response = (
+                operation.first_response_latency_ns is not None
+                or operation.op_type != OperationType.READ_ROWS
+            )
             encountered_exc = None
             try:
                 async for response in call:

google/cloud/bigtable/data/_helpers.py

@@ -123,7 +123,10 @@ def _retry_exception_factory(
 
 def _tracked_exception_factory(
     operation: "ActiveOperationMetric",
-) -> Callable[[list[Exception], RetryFailureReason, float | None], tuple[Exception, Exception | None]]:
+) -> Callable[
+    [list[Exception], RetryFailureReason, float | None],
+    tuple[Exception, Exception | None],
+]:
     """
     wraps and extends _retry_exception_factory to add client-side metrics tracking.
 
@@ -143,19 +146,28 @@ def _tracked_exception_factory(
         tuple[Exception, Exception|None]:
             tuple of the exception to raise, and a cause exception if applicabl
     """
-    def wrapper(exc_list: list[Exception], reason: RetryFailureReason, timeout_val: float | None) -> tuple[Exception, Exception | None]:
+
+    def wrapper(
+        exc_list: list[Exception], reason: RetryFailureReason, timeout_val: float | None
+    ) -> tuple[Exception, Exception | None]:
         source_exc, cause_exc = _retry_exception_factory(exc_list, reason, timeout_val)
         try:
             # record metadata from failed rpc
-            if isinstance(source_exc, core_exceptions.GoogleAPICallError) and source_exc.errors:
+            if (
+                isinstance(source_exc, core_exceptions.GoogleAPICallError)
+                and source_exc.errors
+            ):
                 rpc_error = source_exc.errors[-1]
-                metadata = list(rpc_error.trailing_metadata()) + list(rpc_error.initial_metadata())
-                operation.add_response_metadata({k:v for k,v in metadata})
+                metadata = list(rpc_error.trailing_metadata()) + list(
+                    rpc_error.initial_metadata()
+                )
+                operation.add_response_metadata({k: v for k, v in metadata})
         except Exception:
             # ignore errors in metadata collection
             pass
         operation.end_with_status(source_exc)
         return source_exc, cause_exc
+
     return wrapper
 
 

tests/system/data/__init__.py

@@ -15,12 +15,12 @@
 #
 import pytest
 import uuid
-from google.cloud.bigtable.data._cross_sync import CrossSync
 
 TEST_FAMILY = "test-family"
 TEST_FAMILY_2 = "test-family-2"
 TEST_AGGREGATE_FAMILY = "test-aggregate-family"
 
+
 class SystemTestRunner:
     """
     configures a system test class with configuration for clusters/tables/etc
@@ -67,4 +67,4 @@ def column_family_config(self):
             TEST_AGGREGATE_FAMILY: types.ColumnFamily(
                 value_type=types.Type(aggregate_type=int_aggregate_type)
             ),
-        }
+        }