ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-01-31 09:06:52 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Support running code from module (i.e. '-m' in execution entry point)

Browse Source
This commit is contained in:
allegroai 2020-05-31 13:54:13 +03:00
parent fffe8e1c3f
commit 799b292146
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
trains_agent/commands/worker.py
View File

@ -1294,7 +1294,18 @@ class Worker(ServiceCommandSection):
extra.append(
WorkerParams(optimization=optimization).get_optimization_flag()
)
# check if this is a module load, then load it.
try:
if current_task.script.binary and current_task.script.binary.startswith('python') and \
execution.entry_point and execution.entry_point.split()[0].strip() == '-m':
# we need to split it
import shlex
extra.extend(shlex.split(execution.entry_point))
else:
extra.append(execution.entry_point)
except:
extra.append(execution.entry_point)
command = self.package_api.get_python_command(extra)
print("[{}]$ {}".format(execution.working_dir, command.pretty()))