mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-05 15:28:27 +01:00
Saugat PachhaiandDave Berenbaum
978d5ac227
* exp init: add simple dvc exp init command * Apply suggestions from code review Co-authored-by: Dave Berenbaum <[email protected]> Co-authored-by: Dave Berenbaum <[email protected]>
35 lines
987 B
Python
35 lines
987 B
Python
import os
|
|
|
|
from dvc.command.experiments import CmdExperimentsInit
|
|
from dvc.main import main
|
|
from dvc.utils.serialize import load_yaml
|
|
|
|
|
|
def test_init(tmp_dir, dvc):
|
|
tmp_dir.gen(
|
|
{
|
|
CmdExperimentsInit.CODE: {"copy.py": ""},
|
|
"data": "data",
|
|
"params.yaml": '{"foo": 1}',
|
|
"dvclive": {},
|
|
"plots": {},
|
|
}
|
|
)
|
|
code_path = os.path.join(CmdExperimentsInit.CODE, "copy.py")
|
|
script = f"python {code_path}"
|
|
|
|
assert main(["exp", "init", script]) == 0
|
|
assert load_yaml(tmp_dir / "dvc.yaml") == {
|
|
"stages": {
|
|
"default": {
|
|
"cmd": script,
|
|
"deps": ["data", "src"],
|
|
"live": {"dvclive": {"html": True, "summary": True}},
|
|
"metrics": [{"metrics.json": {"cache": False}}],
|
|
"outs": ["models"],
|
|
"params": ["foo"],
|
|
"plots": [{"plots": {"cache": False}}],
|
|
}
|
|
}
|
|
}
|