ClearML/clearml-session
1
0
Fork 0
mirror of https://github.com/clearml/clearml-session synced 2025-02-24 21:03:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Change default for --keepalive (default is now False)

Browse Source
This commit is contained in:
allegroai 2021-12-13 00:58:22 +02:00
parent 9bd389e7f2
commit 6a85198636
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
clearml_session/__main__.py
View File

@ -723,7 +723,7 @@ def monitor_ssh_tunnel(state, task):
vscode_port = None vscode_port = None
connect_state = {'reconnect': False} connect_state = {'reconnect': False}
if not state.get('disable_keepalive'): if state.get('keepalive'):
if state.get('jupyter_lab'): if state.get('jupyter_lab'):
SingleThreadProxy(local_jupyter_port, local_jupyter_port_) SingleThreadProxy(local_jupyter_port, local_jupyter_port_)
if state.get('vscode_server'): if state.get('vscode_server'):
@ -760,7 +760,7 @@ def monitor_ssh_tunnel(state, task):
ssh_port = \ ssh_port = \
task_parameters.get('properties/k8s-pod-port') or \ task_parameters.get('properties/k8s-pod-port') or \
task_parameters.get('properties/external_ssh_port') or internal_ssh_port task_parameters.get('properties/external_ssh_port') or internal_ssh_port
if not state.get('disable_keepalive'): if state.get('keepalive'):
internal_ssh_port = task_parameters.get('properties/internal_stable_ssh_port') or internal_ssh_port internal_ssh_port = task_parameters.get('properties/internal_stable_ssh_port') or internal_ssh_port
local_remote_pair_list = [(local_ssh_port_, internal_ssh_port)] local_remote_pair_list = [(local_ssh_port_, internal_ssh_port)]
if state.get('jupyter_lab'): if state.get('jupyter_lab'):
@ -931,9 +931,10 @@ def setup_parser(parser):
'(default: previously used Task). Use `none` for the default interactive session') '(default: previously used Task). Use `none` for the default interactive session')
parser.add_argument('--project', type=str, default=None, parser.add_argument('--project', type=str, default=None,
help='Advanced: Set the project name for the interactive session Task') help='Advanced: Set the project name for the interactive session Task')
parser.add_argument('--disable-keepalive', action='store_true', default=None, parser.add_argument('--keepalive', default=False, nargs='?', const='true', metavar='true/false',
help='Advanced: If set, disable the transparent proxy always keeping the sockets alive. ' type=lambda x: (str(x).strip().lower() in ('true', 'yes')),
'Default: false, use transparent socket mitigating connection drops.') help='Advanced: If set, enables the transparent proxy always keeping the sockets alive. '
'Default: False, do not use transparent socket for mitigating connection drops.')
parser.add_argument('--queue-excluded-tag', default=None, nargs='*', parser.add_argument('--queue-excluded-tag', default=None, nargs='*',
help='Advanced: Excluded queues with this specific tag from the selection') help='Advanced: Excluded queues with this specific tag from the selection')
parser.add_argument('--queue-include-tag', default=None, nargs='*', parser.add_argument('--queue-include-tag', default=None, nargs='*',