make_dvcyaml: Include metrics section also if live.summary. (#512)

Author: daavoo

src/dvclive/dvc.py

@@ -89,7 +89,7 @@ def make_dvcyaml(live):
     dvcyaml = {}
     if live._params:
         dvcyaml["params"] = [os.path.relpath(live.params_file, live.dir)]
-    if live._metrics:
+    if live._metrics or live.summary:
         dvcyaml["metrics"] = [os.path.relpath(live.metrics_file, live.dir)]
     plots = []
     plots_path = Path(live.plots_dir)

tests/test_dvc.py

@@ -49,6 +49,16 @@ def test_make_dvcyaml_metrics(tmp_dir):
     }
 
 
+def test_make_dvcyaml_summary(tmp_dir):
+    live = Live()
+    live.summary["bar"] = 2
+    make_dvcyaml(live)
+
+    assert load_yaml(live.dvc_file) == {
+        "metrics": ["metrics.json"],
+    }
+
+
 def test_make_dvcyaml_all_plots(tmp_dir):
     live = Live()
     live.log_param("foo", 1)