diff --git a/openfl/interface/aggregator.py b/openfl/interface/aggregator.py index 9c145e2870..60bd874ee9 100644 --- a/openfl/interface/aggregator.py +++ b/openfl/interface/aggregator.py @@ -52,13 +52,7 @@ def aggregator(context): type=ClickPath(exists=True), ) def start_(plan, authorized_cols): - """ - Start the aggregator service. - - Args: - plan (str): Path to the federated learning plan. - authorized_cols (str): Path to the authorized collaborator list. - """ + """Start the aggregator service.""" if is_directory_traversal(plan): echo("Federated learning plan path is out of the openfl workspace scope.") @@ -154,12 +148,7 @@ def find_certificate_name(file_name): ) @option("-s", "--silent", help="Do not prompt", is_flag=True) def _certify(fqdn, silent): - """Sign/certify the aggregator certificate key pair. - - Args: - fqdn (str): The fully qualified domain name of aggregator node. - silent (bool): Flag to enable silent mode. - """ + """Sign/certify the aggregator certificate key pair.""" certify(fqdn, silent) diff --git a/openfl/interface/collaborator.py b/openfl/interface/collaborator.py index 88722fb174..9bad1e9716 100644 --- a/openfl/interface/collaborator.py +++ b/openfl/interface/collaborator.py @@ -64,13 +64,7 @@ def collaborator(context): help="The certified common name of the collaborator", ) def start_(plan, collaborator_name, data_config): - """Start a collaborator service. - - Args: - plan (str): Federated learning plan. - collaborator_name (str): The certified common name of the collaborator. - data_config (str): The data set/shard configuration file. - """ + """Start a collaborator service.""" if plan and is_directory_traversal(plan): echo("Federated learning plan path is out of the openfl workspace scope.") @@ -337,16 +331,7 @@ def register_collaborator(file_name): help="Import the archive containing the collaborator's" " certificate (signed by the CA)", ) def certify_(collaborator_name, silent, request_pkg, import_): - """Certify the collaborator. - - Args: - collaborator_name (str): The certified common name of the collaborator. - silent (bool): Do not prompt. - request_pkg (str): The archive containing the certificate signing - request (*.zip) for a collaborator. - import_ (str): Import the archive containing the collaborator's - certificate (signed by the CA). - """ + """Certify the collaborator.""" certify(collaborator_name, silent, request_pkg, import_)