fixed lint

Author: daniel-sanche

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

@@ -75,4 +75,4 @@ async def _streaming_generator_wrapper(call):
                 yield response
         except Exception as e:
             # handle errors while processing stream
-            raise
+            raise e

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

@@ -27,21 +27,33 @@ class BigtableMetricsInterceptor(
     """
 
     def intercept_unary_unary(self, continuation, client_call_details, request):
+        """Interceptor for unary rpcs:
+        - MutateRow
+        - CheckAndMutateRow
+        - ReadModifyWriteRow"""
         try:
             call = continuation(client_call_details, request)
             return call
         except Exception as rpc_error:
             raise rpc_error
 
     def intercept_unary_stream(self, continuation, client_call_details, request):
-        def response_wrapper(call):
-            try:
-                for response in call:
-                    yield response
-            except Exception as e:
-                raise e
-
+        """Interceptor for streaming rpcs:
+        - ReadRows
+        - MutateRows
+        - SampleRowKeys"""
         try:
-            return response_wrapper(continuation(client_call_details, request))
+            return self._streaming_generator_wrapper(
+                continuation(client_call_details, request)
+            )
         except Exception as rpc_error:
             raise rpc_error
+
+    @staticmethod
+    def _streaming_generator_wrapper(call):
+        """Wrapped generator to be returned by intercept_unary_stream"""
+        try:
+            for response in call:
+                yield response
+        except Exception as e:
+            raise e