From d2812fa48fc8a50eae3d7f4f0ce6556c0aa3f552 Mon Sep 17 00:00:00 2001 From: David de la Iglesia Castro Date: Mon, 27 Mar 2023 20:29:00 +0200 Subject: [PATCH] studio: Don't fail on empty params. --- dvc/repo/experiments/utils.py | 2 +- tests/unit/repo/experiments/test_utils.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/dvc/repo/experiments/utils.py b/dvc/repo/experiments/utils.py index afd1a1a5a..c7106caa3 100644 --- a/dvc/repo/experiments/utils.py +++ b/dvc/repo/experiments/utils.py @@ -340,6 +340,6 @@ def to_studio_params(dvc_params): return result for rev_data in dvc_params.values(): for file_name, file_data in rev_data.get("data", {}).items(): - result[file_name] = file_data["data"] + result[file_name] = file_data.get("data", {}) return result diff --git a/tests/unit/repo/experiments/test_utils.py b/tests/unit/repo/experiments/test_utils.py index 1e93321fe..7b2c6c182 100644 --- a/tests/unit/repo/experiments/test_utils.py +++ b/tests/unit/repo/experiments/test_utils.py @@ -62,6 +62,10 @@ def test_run_check_ref_format(scm, name, result): {"workspace": {"data": {"params.yaml": {"data": {"foo": 1}}}}}, {"params.yaml": {"foo": 1}}, ), + ( + {"workspace": {"data": {"params.yaml": {"error": "FileNotFound"}}}}, + {"params.yaml": {}}, + ), ( {"workspace": {"error": "something went wrong"}}, {},