From a4295073129709ed35e82ec3a5ebacfdeb63852c Mon Sep 17 00:00:00 2001 From: Jina Dev Bot Date: Thu, 30 Nov 2023 11:07:26 +0000 Subject: [PATCH] style: fix overload and cli autocomplete --- jina/orchestrate/deployments/__init__.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/jina/orchestrate/deployments/__init__.py b/jina/orchestrate/deployments/__init__.py index fed613cc7a02c..e688fba8036a5 100644 --- a/jina/orchestrate/deployments/__init__.py +++ b/jina/orchestrate/deployments/__init__.py @@ -106,6 +106,7 @@ async def _async_call_add_voters(leader, voters, replica_ids, logger): # this method needs to be run in multiprocess, importing jraft in main process # makes it impossible to do tests sequentially from jina.serve.consensus.add_voter.call_add_voter import async_call_add_voter + logger.debug(f'Trying to add {len(replica_ids)} voters to leader {leader}') for voter_address, replica_id in zip(voters, replica_ids): logger.debug( @@ -114,7 +115,9 @@ async def _async_call_add_voters(leader, voters, replica_ids, logger): success = False for i in range(5): logger.debug(f'Trying {i}th time') - success = await async_call_add_voter(leader, str(replica_id), voter_address, logger) + success = await async_call_add_voter( + leader, str(replica_id), voter_address, logger + ) if success: logger.debug(f'Trying {i}th time succeeded') break @@ -134,6 +137,7 @@ async def _async_call_add_voters(leader, voters, replica_ids, logger): logger.debug('Adding voters to leader finished') return success + class Deployment(JAMLCompatible, PostMixin, BaseOrchestrator, metaclass=DeploymentType): """A Deployment is an immutable set of pods, which run in replicas. They share the same input and output socket. Internally, the pods can run with the process/thread backend. They can also be run in their own containers @@ -181,11 +185,11 @@ async def _async_add_voter_to_leader(self): replica_ids = [pod.args.replica_id for pod in self._pods[1:]] self.logger.debug('Starting process to call Add Voters') res = await _async_call_add_voters( - leader=leader_address, - voters=voter_addresses, - replica_ids=replica_ids, - logger=self.logger, - ) + leader=leader_address, + voters=voter_addresses, + replica_ids=replica_ids, + logger=self.logger, + ) if res: self.logger.debug('Add Voters process finished') else: