ran format

Author: daniel-sanche

packages/google-cloud-bigtable/google/cloud/bigtable/data/_async/client.py

@@ -70,7 +70,7 @@
 )
 from google.cloud.bigtable.data._metrics.handlers.gcp_exporter import (
     BigtableMetricsExporter,
-    GoogleCloudMetricsHandler
+    GoogleCloudMetricsHandler,
 )
 from google.cloud.bigtable.data.exceptions import (
     FailedQueryShardError,

packages/google-cloud-bigtable/google/cloud/bigtable/data/_metrics/__init__.py

@@ -11,12 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from google.cloud.bigtable.data._metrics.handlers.opentelemetry import (
-    OpenTelemetryMetricsHandler,
-)
-from google.cloud.bigtable.data._metrics.handlers.gcp_exporter import (
-    GoogleCloudMetricsHandler,
-)
 from google.cloud.bigtable.data._metrics.data_model import (
     ActiveAttemptMetric,
     ActiveOperationMetric,
@@ -25,6 +19,12 @@
     OperationState,
     OperationType,
 )
+from google.cloud.bigtable.data._metrics.handlers.gcp_exporter import (
+    GoogleCloudMetricsHandler,
+)
+from google.cloud.bigtable.data._metrics.handlers.opentelemetry import (
+    OpenTelemetryMetricsHandler,
+)
 from google.cloud.bigtable.data._metrics.metrics_controller import (
     BigtableClientSideMetricsController,
 )

packages/google-cloud-bigtable/google/cloud/bigtable/data/_metrics/handlers/gcp_exporter.py

@@ -16,21 +16,10 @@
 
 import time
 
-from opentelemetry.sdk.metrics import MeterProvider
-from opentelemetry.sdk.metrics import view
-from opentelemetry.sdk.metrics.export import (
-    HistogramDataPoint,
-    MetricExporter,
-    MetricExportResult,
-    MetricsData,
-    NumberDataPoint,
-    PeriodicExportingMetricReader,
-)
-from google.protobuf.timestamp_pb2 import Timestamp
 from google.api.distribution_pb2 import Distribution
 from google.api.metric_pb2 import Metric as GMetric
-from google.api.monitored_resource_pb2 import MonitoredResource
 from google.api.metric_pb2 import MetricDescriptor
+from google.api.monitored_resource_pb2 import MonitoredResource
 from google.api_core import gapic_v1
 from google.cloud.monitoring_v3 import (
     CreateTimeSeriesRequest,
@@ -40,15 +29,22 @@
     TimeSeries,
     TypedValue,
 )
+from google.protobuf.timestamp_pb2 import Timestamp
+from opentelemetry.sdk.metrics import MeterProvider, view
+from opentelemetry.sdk.metrics.export import (
+    HistogramDataPoint,
+    MetricExporter,
+    MetricExportResult,
+    MetricsData,
+    NumberDataPoint,
+    PeriodicExportingMetricReader,
+)
 
 from google.cloud.bigtable.data._metrics.handlers.opentelemetry import (
     OpenTelemetryMetricsHandler,
-)
-from google.cloud.bigtable.data._metrics.handlers.opentelemetry import (
     _OpenTelemetryInstruments,
 )
 
-
 # create OpenTelemetry views for Bigtable metrics
 # avoid reformatting into individual lines
 # fmt: off

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

@@ -18,13 +18,15 @@
 import uuid
 
 from google.cloud.bigtable import __version__ as bigtable_version
+from google.cloud.bigtable.data._metrics.data_model import (
+    DEFAULT_CLUSTER_ID,
+    DEFAULT_ZONE,
+    ActiveOperationMetric,
+    CompletedAttemptMetric,
+    CompletedOperationMetric,
+    OperationType,
+)
 from google.cloud.bigtable.data._metrics.handlers._base import MetricsHandler
-from google.cloud.bigtable.data._metrics.data_model import OperationType
-from google.cloud.bigtable.data._metrics.data_model import DEFAULT_CLUSTER_ID
-from google.cloud.bigtable.data._metrics.data_model import DEFAULT_ZONE
-from google.cloud.bigtable.data._metrics.data_model import ActiveOperationMetric
-from google.cloud.bigtable.data._metrics.data_model import CompletedAttemptMetric
-from google.cloud.bigtable.data._metrics.data_model import CompletedOperationMetric
 
 # conversion factor for converting from nanoseconds to milliseconds
 NS_TO_MS = 1e6
@@ -216,9 +218,7 @@ def on_attempt_complete(
             {"streaming": is_streaming, "status": status, **labels},
         )
         flow_throttling = (
-            op.flow_throttling_time_ns / NS_TO_MS
-            if op.flow_throttling_time_ns
-            else 0
+            op.flow_throttling_time_ns / NS_TO_MS if op.flow_throttling_time_ns else 0
         )
         self.otel.throttling_latencies.record(flow_throttling, labels)
         self.otel.application_latencies.record(

packages/google-cloud-bigtable/tests/system/data/__init__.py

@@ -13,9 +13,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
+import datetime
 import os
 import uuid
-import datetime
 
 import pytest
 

packages/google-cloud-bigtable/tests/unit/data/_async/test_client.py

@@ -115,6 +115,7 @@ async def test_ctor(self):
         from google.cloud.bigtable.data._metrics.handlers.gcp_exporter import (
             BigtableMetricsExporter,
         )
+
         expected_project = "project-id"
         expected_credentials = AnonymousCredentials()
         client = self._make_client(
@@ -1210,8 +1211,8 @@ async def test_ctor(self):
         from google.cloud.bigtable.data._helpers import _WarmedInstanceKey
         from google.cloud.bigtable.data._metrics import (
             BigtableClientSideMetricsController,
+            GoogleCloudMetricsHandler,
         )
-        from google.cloud.bigtable.data._metrics import GoogleCloudMetricsHandler
 
         expected_table_id = "table-id"
         expected_instance_id = "instance-id"
@@ -1665,6 +1666,7 @@ def _make_table(self, *args, **kwargs):
         from google.cloud.bigtable.data._metrics.handlers.gcp_exporter import (
             BigtableMetricsExporter,
         )
+
         client_mock = mock.Mock()
         client_mock._register_instance.side_effect = (
             lambda *args, **kwargs: CrossSync.yield_to_event_loop()

packages/google-cloud-bigtable/tests/unit/data/_metrics/test_gcp_exporter_handler.py

@@ -11,32 +11,28 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import pytest
 import mock
-
+import pytest
+from google.api.distribution_pb2 import Distribution
+from google.cloud.monitoring_v3 import (
+    Point,
+    TimeSeries,
+)
 from opentelemetry.sdk.metrics import MeterProvider
 from opentelemetry.sdk.metrics.export import (
-    NumberDataPoint,
+    AggregationTemporality,
     HistogramDataPoint,
+    Metric,
     MetricExportResult,
     MetricsData,
+    NumberDataPoint,
     ResourceMetrics,
     ScopeMetrics,
-    Metric,
     Sum,
-    AggregationTemporality,
-)
-from google.cloud.monitoring_v3 import (
-    Point,
-    TimeSeries,
 )
-from google.api.distribution_pb2 import Distribution
-
 
 from google.cloud.bigtable.data._metrics.handlers.gcp_exporter import (
     BigtableMetricsExporter,
-)
-from google.cloud.bigtable.data._metrics.handlers.gcp_exporter import (
     GoogleCloudMetricsHandler,
 )
 from google.cloud.bigtable.data._metrics.handlers.opentelemetry import (
@@ -284,6 +280,7 @@ def test__batch_write(
     def test__batch_write_with_deadline(self):
         """Test that _batch_write passes deadlines to gapic correctly."""
         import time
+
         from google.api_core import gapic_v1
 
         instance = self._make_one("project")

packages/google-cloud-bigtable/tests/unit/data/_metrics/test_opentelemetry_handler.py

@@ -11,9 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import pytest
 import mock
-
+import pytest
 from grpc import StatusCode
 
 from google.cloud.bigtable.data._metrics.data_model import (
@@ -23,8 +22,8 @@
     OperationType,
 )
 from google.cloud.bigtable.data._metrics.handlers.opentelemetry import (
-    _OpenTelemetryInstruments,
     OpenTelemetryMetricsHandler,
+    _OpenTelemetryInstruments,
 )