LINT

Signed-off-by: Qing Wang <kingchin1218@gmail.com>
ray-project · Jul 4, 2023 · 4ec3360 · 4ec3360
1 parent c49f22e
commit 4ec3360
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/benchmarks/many_tiny_tasks_benchmark.py b/benchmarks/many_tiny_tasks_benchmark.py
@@ -15,7 +15,6 @@
 import ray
 import time
 import sys
-import ray
 import fed
 
 

diff --git a/fed/proxy/barriers.py b/fed/proxy/barriers.py
@@ -139,7 +139,10 @@ async def send_data_grpc(
     )
     # Waiting for the reply from downstream.
     response = await stub.SendData(
-        request, metadata=fed_utils.dict2tuple(metadata), timeout=cluster_config.cross_silo_timeout)
+        request,
+        metadata=fed_utils.dict2tuple(metadata),
+        timeout=cluster_config.cross_silo_timeout,
+    )
     logger.debug(
         f'Received data response from seq_id {downstream_seq_id}, '
         f'result: {response.result}.'
@@ -200,13 +203,14 @@ async def send(
         dest_party_grpc_config = self.setup_grpc_config(dest_party)
         try:
             tls_enabled = fed_utils.tls_enabled(self._tls_config)
-            grpc_options =dest_party_grpc_config['grpc_options']
+            grpc_options = dest_party_grpc_config['grpc_options']
             grpc_options = get_grpc_options(retry_policy=self.retry_policy) if \
-                            grpc_options is None else fed_utils.dict2tuple(grpc_options)
-            
+                grpc_options is None else fed_utils.dict2tuple(grpc_options)
+
             if dest_party not in self._stubs:
                 if tls_enabled:
-                    ca_cert, private_key, cert_chain = fed_utils.load_cert_config(self._tls_config)
+                    ca_cert, private_key, cert_chain = fed_utils.load_cert_config(
+                        self._tls_config)
                     credentials = grpc.ssl_channel_credentials(
                         certificate_chain=cert_chain,
                         private_key=private_key,