mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-12 07:18:23 +01:00
skshetry
9a1bcd7f1d
The code are wrapped due to us using line-length=79 before, but now some of the code can easily fit in 88 chars. The formatter did not change this automatically when we changed line-length, because of magic trailing commas that gets added when they turn into multiple lines. Black/ruff wrapping them in parens and splitting them into multiple lines was making it difficult to read.
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
from dvc.cli import formatter
|
|
from dvc.cli.command import CmdBase
|
|
from dvc.cli.utils import append_doc_link
|
|
from dvc.log import logger
|
|
|
|
logger = logger.getChild(__name__)
|
|
|
|
|
|
class CmdQueueLogs(CmdBase):
|
|
"""Show output logs for a queued experiment."""
|
|
|
|
def run(self):
|
|
self.repo.experiments.celery_queue.logs(
|
|
rev=self.args.task,
|
|
encoding=self.args.encoding,
|
|
follow=self.args.follow,
|
|
)
|
|
|
|
return 0
|
|
|
|
|
|
def add_parser(queue_subparsers, parent_parser):
|
|
QUEUE_LOGS_HELP = (
|
|
"Show output logs for running and completed experiment queue tasks."
|
|
)
|
|
queue_logs_parser = queue_subparsers.add_parser(
|
|
"logs",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(QUEUE_LOGS_HELP, "queue/logs"),
|
|
help=QUEUE_LOGS_HELP,
|
|
formatter_class=formatter.RawDescriptionHelpFormatter,
|
|
)
|
|
queue_logs_parser.add_argument(
|
|
"-e",
|
|
"--encoding",
|
|
help="Text encoding for log output. Defaults to system locale encoding.",
|
|
metavar="<encoding>",
|
|
)
|
|
queue_logs_parser.add_argument(
|
|
"-f",
|
|
"--follow",
|
|
help=(
|
|
"Attach to task and follow additional live output. Only "
|
|
"applicable if the task is still running."
|
|
),
|
|
action="store_true",
|
|
)
|
|
queue_logs_parser.add_argument("task", help="Task to show.", metavar="<task>")
|
|
queue_logs_parser.set_defaults(func=CmdQueueLogs)
|