mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-09 07:18:27 +01:00
David de la Iglesia Castro
c74dc20b97
Custom commit message to use when committing the experiment. Part of #8870
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
import logging
|
|
|
|
from dvc.cli.command import CmdBaseNoRepo
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class CmdExecutorRun(CmdBaseNoRepo):
|
|
"""Run an experiment executor."""
|
|
|
|
def run(self):
|
|
from dvc.repo.experiments.executor.base import BaseExecutor, ExecutorInfo
|
|
from dvc.utils.serialize import load_json
|
|
|
|
info = ExecutorInfo.from_dict(load_json(self.args.infofile))
|
|
BaseExecutor.reproduce(
|
|
info=info,
|
|
rev="",
|
|
queue=None,
|
|
log_level=logger.getEffectiveLevel(),
|
|
infofile=self.args.infofile,
|
|
copy_paths=self.args.copy_paths,
|
|
message=self.args.message,
|
|
)
|
|
return 0
|
|
|
|
|
|
def add_parser(experiments_subparsers, parent_parser):
|
|
EXEC_RUN_HELP = "Run an experiment executor."
|
|
exec_run_parser = experiments_subparsers.add_parser(
|
|
"exec-run",
|
|
parents=[parent_parser],
|
|
description=EXEC_RUN_HELP,
|
|
add_help=False,
|
|
)
|
|
exec_run_parser.add_argument(
|
|
"--infofile",
|
|
help="Path to executor info file",
|
|
default=None,
|
|
)
|
|
exec_run_parser.add_argument(
|
|
"-C",
|
|
"--copy-paths",
|
|
action="append",
|
|
default=[],
|
|
help=(
|
|
"List of ignored or untracked paths to copy into the temp directory."
|
|
" Only used if `--temp` or `--queue` is specified."
|
|
),
|
|
)
|
|
exec_run_parser.add_argument(
|
|
"-M",
|
|
"--message",
|
|
type=str,
|
|
default=None,
|
|
help="Custom commit message to use when committing the experiment.",
|
|
)
|
|
exec_run_parser.set_defaults(func=CmdExecutorRun)
|