optuna: Use summary instead of log_metric. (#514)

Prevent creation of single-row `.tsv` files. Each trial creates a single DVC experiment and logs metrics a single time.
iterative · Apr 3, 2023 · c9fe84d · c9fe84d
1 parent 634f911
commit c9fe84d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/dvclive/optuna.py b/src/dvclive/optuna.py
@@ -43,4 +43,4 @@ def _log_metrics(self, values, live):
 
         metrics = {name: val for name, val in zip(names, values)}
         for k, v in metrics.items():
-            live.log_metric(k, v)
+            live.summary[k] = v
diff --git a/tests/test_frameworks/test_optuna.py b/tests/test_frameworks/test_optuna.py
@@ -25,3 +25,5 @@ def test_optuna_(tmp_dir, mocked_dvc_repo):
     assert metric_name in metrics
     params = load_yaml("dvclive-optuna/params.yaml")
     assert "x" in params
+
+    assert not (tmp_dir / "dvclive-optuna" / "plots").exists()