20
20
from aiida .cmdline .params import arguments , options
21
21
from aiida .cmdline .params .options .commands import computer as options_computer
22
22
from 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
24
24
from aiida .cmdline .utils .decorators import with_dbenv
25
25
from aiida .common .exceptions import EntryPointError , ValidationError
26
26
from aiida .plugins .entry_point import get_entry_point_names
@@ -860,7 +860,7 @@ def computer_search(ctx, pattern, source):
860
860
ComputerSelector ,
861
861
ComputerSetupHandler ,
862
862
ComputerSource ,
863
- ) # noqa: PLC0415
863
+ )
864
864
865
865
# Convert string to enum
866
866
source_enum = ComputerSource (source )
@@ -895,10 +895,13 @@ def computer_search(ctx, pattern, source):
895
895
# Handle SSH-only source differently
896
896
if source_enum == ComputerSource .SSH_CONFIG :
897
897
# 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
+ )
902
905
echo .echo_report ('Computers registered in the ~/.ssh/config can be set up using the `core.ssh_async` transport' )
903
906
echo .echo_report ('This transport plugin automatically uses your OS SSH configuration' )
904
907
echo .echo_report ('for connection settings including ProxyJump, IdentityFile, and other SSH options.' )
@@ -919,7 +922,9 @@ def computer_search(ctx, pattern, source):
919
922
return
920
923
921
924
# 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
+ ):
923
928
print ('Setup cancelled.' )
924
929
return
925
930
0 commit comments