@@ -326,13 +326,16 @@ def _get_state_and_remaining_time(
326
326
@overload
327
327
def _get_controlling_quota (
328
328
self , state : Literal [CircuitBreakerState .OK , CircuitBreakerState .RECOVERY ]
329
- ) -> Quota : ...
329
+ ) -> Quota :
330
+ ...
330
331
331
332
@overload
332
- def _get_controlling_quota (self , state : Literal [CircuitBreakerState .BROKEN ]) -> None : ...
333
+ def _get_controlling_quota (self , state : Literal [CircuitBreakerState .BROKEN ]) -> None :
334
+ ...
333
335
334
336
@overload
335
- def _get_controlling_quota (self ) -> Quota | None : ...
337
+ def _get_controlling_quota (self ) -> Quota | None :
338
+ ...
336
339
337
340
def _get_controlling_quota (self , state : CircuitBreakerState | None = None ) -> Quota | None :
338
341
"""
@@ -350,19 +353,24 @@ def _get_controlling_quota(self, state: CircuitBreakerState | None = None) -> Qu
350
353
return controlling_quota_by_state [_state ]
351
354
352
355
@overload
353
- def _get_remaining_error_quota (self , quota : None , window_end : int | None ) -> None : ...
356
+ def _get_remaining_error_quota (self , quota : None , window_end : int | None ) -> None :
357
+ ...
354
358
355
359
@overload
356
- def _get_remaining_error_quota (self , quota : Quota , window_end : int | None ) -> int : ...
360
+ def _get_remaining_error_quota (self , quota : Quota , window_end : int | None ) -> int :
361
+ ...
357
362
358
363
@overload
359
- def _get_remaining_error_quota (self , quota : None ) -> None : ...
364
+ def _get_remaining_error_quota (self , quota : None ) -> None :
365
+ ...
360
366
361
367
@overload
362
- def _get_remaining_error_quota (self , quota : Quota ) -> int : ...
368
+ def _get_remaining_error_quota (self , quota : Quota ) -> int :
369
+ ...
363
370
364
371
@overload
365
- def _get_remaining_error_quota (self ) -> int | None : ...
372
+ def _get_remaining_error_quota (self ) -> int | None :
373
+ ...
366
374
367
375
def _get_remaining_error_quota (
368
376
self , quota : Quota | None = None , window_end : int | None = None
0 commit comments