Skip to content

Commit a548fef

Browse files
committed
fix formatting
1 parent fa30212 commit a548fef

File tree

3 files changed

+10
-14
lines changed

3 files changed

+10
-14
lines changed

sdks/python/apache_beam/examples/rate_limiter_simple.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,9 @@
2424
import time
2525

2626
import apache_beam as beam
27-
from apache_beam.utils import shared
2827
from apache_beam.io.components.rate_limiter import EnvoyRateLimiter
2928
from apache_beam.options.pipeline_options import PipelineOptions
29+
from apache_beam.utils import shared
3030

3131

3232
class SampleApiDoFn(beam.DoFn):

sdks/python/apache_beam/io/components/rate_limiter.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -28,10 +28,8 @@
2828
from typing import List
2929

3030
import grpc
31-
from envoy_data_plane.envoy.extensions.common.ratelimit.v3 import (
32-
RateLimitDescriptor)
33-
from envoy_data_plane.envoy.extensions.common.ratelimit.v3 import (
34-
RateLimitDescriptorEntry)
31+
from envoy_data_plane.envoy.extensions.common.ratelimit.v3 import RateLimitDescriptor
32+
from envoy_data_plane.envoy.extensions.common.ratelimit.v3 import RateLimitDescriptorEntry
3533
from envoy_data_plane.envoy.service.ratelimit.v3 import RateLimitRequest
3634
from envoy_data_plane.envoy.service.ratelimit.v3 import RateLimitResponse
3735
from envoy_data_plane.envoy.service.ratelimit.v3 import RateLimitResponseCode
@@ -52,16 +50,14 @@ def __init__(self, namespace: str = ""):
5250
# Metric updates are thread safe
5351
self.throttling_signaler = adaptive_throttler.ThrottlingSignaler(
5452
namespace=namespace)
55-
self.requests_counter = Metrics.counter(
56-
namespace, 'RatelimitRequestsTotal')
53+
self.requests_counter = Metrics.counter(namespace, 'RatelimitRequestsTotal')
5754
self.requests_allowed = Metrics.counter(
5855
namespace, 'RatelimitRequestsAllowed')
5956
self.requests_throttled = Metrics.counter(
6057
namespace, 'RatelimitRequestsThrottled')
6158
self.rpc_errors = Metrics.counter(namespace, 'RatelimitRpcErrors')
6259
self.rpc_retries = Metrics.counter(namespace, 'RatelimitRpcRetries')
63-
self.rpc_latency = Metrics.distribution(
64-
namespace, 'RatelimitRpcLatencyMs')
60+
self.rpc_latency = Metrics.distribution(namespace, 'RatelimitRpcLatencyMs')
6561

6662
@abc.abstractmethod
6763
def throttle(self, **kwargs) -> bool:

sdks/python/apache_beam/io/components/rate_limiter_test.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -16,15 +16,15 @@
1616
#
1717

1818
import unittest
19-
from unittest import mock
20-
import grpc
2119
from datetime import timedelta
20+
from unittest import mock
2221

23-
from apache_beam.io.components import rate_limiter
22+
import grpc
2423
from envoy_data_plane.envoy.service.ratelimit.v3 import RateLimitResponse
2524
from envoy_data_plane.envoy.service.ratelimit.v3 import RateLimitResponseCode
26-
from envoy_data_plane.envoy.service.ratelimit.v3 import (
27-
RateLimitResponseDescriptorStatus)
25+
from envoy_data_plane.envoy.service.ratelimit.v3 import RateLimitResponseDescriptorStatus
26+
27+
from apache_beam.io.components import rate_limiter
2828

2929

3030
class EnvoyRateLimiterTest(unittest.TestCase):

0 commit comments

Comments
 (0)