formatted

armadaproject · Jun 22, 2023 · 019d6fa · 019d6fa
1 parent c9b5a59
commit 019d6fa
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/third_party/airflow/armada/operators/jobservice.py b/third_party/airflow/armada/operators/jobservice.py
@@ -4,6 +4,7 @@
 
 import tenacity
 
+
 class JobServiceClient:
  """
  The JobService Client
@@ -19,7 +20,11 @@ class JobServiceClient:
  def __init__(self, channel):
  self.job_stub = jobservice_pb2_grpc.JobServiceStub(channel)
 
- @tenacity.retry(stop=tenacity.stop_after_attempt(3), wait=tenacity.wait_exponential(), reraise=True)
+ @tenacity.retry(
+ stop=tenacity.stop_after_attempt(3),
+ wait=tenacity.wait_exponential(),
+ reraise=True,
+ )
  def get_job_status(
  self, queue: str, job_set_id: str, job_id: str
  ) -> jobservice_pb2.JobServiceResponse:
@@ -36,8 +41,12 @@ def get_job_status(
  queue=queue, job_set_id=job_set_id, job_id=job_id
  )
  return self.job_stub.GetJobStatus(job_service_request)
-
- @tenacity.retry(stop=tenacity.stop_after_attempt(3), wait=tenacity.wait_exponential(), reraise=True)
+
+ @tenacity.retry(
+ stop=tenacity.stop_after_attempt(3),
+ wait=tenacity.wait_exponential(),
+ reraise=True,
+ )
  def health(self) -> jobservice_pb2.HealthCheckResponse:
  """Health Check for GRPC Request"""
  return self.job_stub.Health(request=empty_pb2.Empty())