File tree Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -429,7 +429,7 @@ def close_tb(logging_dir):
429
429
def refresh_tasks (running_task = None ):
430
430
output_dir = running_task if not running_task or 'pid:' not in running_task else None
431
431
process_name = 'swift'
432
- negative_name = 'swift.exe'
432
+ negative_names = [ 'swift.exe' , 'swift-script.py' ]
433
433
cmd_name = ['pt' , 'sft' , 'rlhf' ]
434
434
process = []
435
435
selected = None
@@ -438,10 +438,12 @@ def refresh_tasks(running_task=None):
438
438
cmdlines = proc .cmdline ()
439
439
except (psutil .ZombieProcess , psutil .AccessDenied , psutil .NoSuchProcess ):
440
440
cmdlines = []
441
- if any ([process_name in cmdline
442
- for cmdline in cmdlines ]) and not any ([negative_name in cmdline
443
- for cmdline in cmdlines ]) and any ( # noqa
444
- [cmdline in cmd_name for cmdline in cmdlines ]): # noqa
441
+ if any ([
442
+ process_name in cmdline for cmdline in cmdlines # noqa
443
+ ]) and not any ([ # noqa
444
+ negative_name in cmdline for negative_name in negative_names # noqa
445
+ for cmdline in cmdlines # noqa
446
+ ]) and any ([cmdline in cmd_name for cmdline in cmdlines ]): # noqa
445
447
process .append (Runtime .construct_running_task (proc ))
446
448
if output_dir is not None and any ( # noqa
447
449
[output_dir == cmdline for cmdline in cmdlines ]): # noqa
You can’t perform that action at this time.
0 commit comments