mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-09 23:10:24 +01:00
92 lines
2.5 KiB
Python
92 lines
2.5 KiB
Python
import argparse
|
|
import os
|
|
|
|
from dvc.command import completion
|
|
from dvc.command.base import CmdBase, fix_subparsers
|
|
from dvc.ui import ui
|
|
|
|
|
|
class CmdLive(CmdBase):
|
|
UNINITIALIZED = True
|
|
|
|
def _run(self, target, revs=None):
|
|
metrics, plots = self.repo.live.show(target=target, revs=revs)
|
|
|
|
html_path = self.args.target + ".html"
|
|
self.repo.plots.write_html(html_path, plots, metrics)
|
|
|
|
ui.write("\nfile://", os.path.abspath(html_path), sep="")
|
|
|
|
return 0
|
|
|
|
|
|
class CmdLiveShow(CmdLive):
|
|
def run(self):
|
|
return self._run(self.args.target)
|
|
|
|
|
|
class CmdLiveDiff(CmdLive):
|
|
def run(self):
|
|
return self._run(self.args.target, self.args.revs)
|
|
|
|
|
|
def shared_parent_parser():
|
|
parent_parser = argparse.ArgumentParser(add_help=False)
|
|
parent_parser.add_argument(
|
|
"target", help="Logs dir to produce summary from"
|
|
).complete = completion.DIR
|
|
parent_parser.add_argument(
|
|
"-o",
|
|
"--out",
|
|
default=None,
|
|
help="Destination path to save plots to",
|
|
metavar="<path>",
|
|
).complete = completion.DIR
|
|
return parent_parser
|
|
|
|
|
|
def add_parser(subparsers, parent_parser):
|
|
LIVE_DESCRIPTION = (
|
|
"Commands to visualize and compare dvclive-produced logs."
|
|
)
|
|
live_parser = subparsers.add_parser(
|
|
"live",
|
|
parents=[parent_parser],
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
description=LIVE_DESCRIPTION,
|
|
)
|
|
live_subparsers = live_parser.add_subparsers(
|
|
dest="cmd",
|
|
help="Use `dvc live CMD --help` to display command-specific help.",
|
|
)
|
|
|
|
fix_subparsers(live_subparsers)
|
|
|
|
SHOW_HELP = "Visualize dvclive directory content."
|
|
live_show_parser = live_subparsers.add_parser(
|
|
"show",
|
|
parents=[parent_parser, shared_parent_parser()],
|
|
help=SHOW_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
live_show_parser.set_defaults(func=CmdLiveShow)
|
|
|
|
DIFF_HELP = (
|
|
"Show multiple versions of dvclive data, "
|
|
"by plotting it in single view."
|
|
)
|
|
live_diff_parser = live_subparsers.add_parser(
|
|
"diff",
|
|
parents=[parent_parser, shared_parent_parser()],
|
|
help=DIFF_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
live_diff_parser.add_argument(
|
|
"--revs",
|
|
nargs="*",
|
|
default=None,
|
|
help="Git revision (e.g. SHA, branch, tag)",
|
|
metavar="<commit>",
|
|
)
|
|
live_diff_parser.set_defaults(func=CmdLiveDiff)
|