2020from aiida .cmdline .params import arguments , options
2121from aiida .cmdline .params .options .commands import computer as options_computer
2222from aiida .cmdline .utils import echo , echo_tabulate
23- from aiida .cmdline .utils .common import validate_output_filename , tabulate
23+ from aiida .cmdline .utils .common import tabulate , validate_output_filename
2424from aiida .cmdline .utils .decorators import with_dbenv
2525from aiida .common .exceptions import EntryPointError , ValidationError
2626from aiida .plugins .entry_point import get_entry_point_names
@@ -860,7 +860,7 @@ def computer_search(ctx, pattern, source):
860860 ComputerSelector ,
861861 ComputerSetupHandler ,
862862 ComputerSource ,
863- ) # noqa: PLC0415
863+ )
864864
865865 # Convert string to enum
866866 source_enum = ComputerSource (source )
@@ -895,10 +895,13 @@ def computer_search(ctx, pattern, source):
895895 # Handle SSH-only source differently
896896 if source_enum == ComputerSource .SSH_CONFIG :
897897 # Show table of SSH config computers
898- print (tabulate (
899- [[i + 1 , k ] for i , k in enumerate (registry_data .keys ())],
900- headers = ['#' , 'SSH Config Computer' ],
901- tablefmt = 'grid' ))
898+ print (
899+ tabulate (
900+ [[i + 1 , k ] for i , k in enumerate (registry_data .keys ())],
901+ headers = ['#' , 'SSH Config Computer' ],
902+ tablefmt = 'grid' ,
903+ )
904+ )
902905 echo .echo_report ('Computers registered in the ~/.ssh/config can be set up using the `core.ssh_async` transport' )
903906 echo .echo_report ('This transport plugin automatically uses your OS SSH configuration' )
904907 echo .echo_report ('for connection settings including ProxyJump, IdentityFile, and other SSH options.' )
@@ -919,7 +922,9 @@ def computer_search(ctx, pattern, source):
919922 return
920923
921924 # Prompt user before setup
922- if not click .confirm (f'Would you like to set up the computer "{ system_name } " with variant "{ variant } " now?' , default = True ):
925+ if not click .confirm (
926+ f'Would you like to set up the computer "{ system_name } " with variant "{ variant } " now?' , default = True
927+ ):
923928 print ('Setup cancelled.' )
924929 return
925930
0 commit comments