mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-07 07:18:27 +01:00
skshetry
519cce3fe4
* use logger.getChild from `dvc.log` instead of `logging.getLogger` I got tired adding `# type: ignore[attr-defined]` while using `logger.trace`. So, now you need to import logger from `from dvc.log import logger` and create a child logger using `logger.getChild(__name__)`. `logger.getChild` will return a logger type-hinted as a logger having a `trace` method. So, from now on, instead of: ```python import logging logger = logging.getogger(__name__) ``` you'll have to do following: ```python from dvc.log import logger logger = logger.getChild(__name__) ``` * remove all dvc imports from dvc.logger * use Tqdm back
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
from dvc.cli.command import CmdBaseNoRepo
|
|
from dvc.log import logger
|
|
|
|
logger = logger.getChild(__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)
|