Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-10 07:18:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

plots: accept any number of y data sources

Browse Source
This commit is contained in:
dberenbaum
2022-12-20 14:32:40 +01:00
committed by David de la Iglesia Castro
parent b4b5db4b4e
commit 6fc0a6ff98
2 changed files with 54 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dvc/render/converter/vega.py
+6 -2
View File
@@ -134,8 +134,12 @@ class VegaConverter(Converter):
self.inferred_properties["x"] = {}
# If multiple y files, duplicate x for each file.
if isinstance(y, dict):
for file in y.keys():
self.inferred_properties["x"][file] = x
for file, fields in y.items():
# Duplicate x for each y.
if isinstance(fields, list):
self.inferred_properties["x"][file] = [x] * len(fields)
else:
self.inferred_properties["x"][file] = x
# Otherwise use plot ID as file.
else:
self.inferred_properties["x"][self.plot_id] = x
tests/unit/render/test_vega_converter.py
+48
View File
@@ -331,6 +331,54 @@ def test_finding_lists(dictionary, expected_result):
},
id="y_list",
),
pytest.param(
{
"f": {"metric": [{"v": 1, "v2": 0.1, "v3": 0.01}]},
"f2": {"metric": [{"v": 1, "v2": 0.1}]},
},
{"y": {"f": ["v2", "v3"], "f2": ["v2"]}, "x": "v"},
[
{
"dvc_inferred_y_value": 0.1,
"v": 1,
"v2": 0.1,
"v3": 0.01,
VERSION_FIELD: {
"revision": "r",
"filename": "f",
"field": "v2",
},
},
{
"dvc_inferred_y_value": 0.01,
"v": 1,
"v2": 0.1,
"v3": 0.01,
VERSION_FIELD: {
"revision": "r",
"filename": "f",
"field": "v3",
},
},
{
"dvc_inferred_y_value": 0.1,
"v": 1,
"v2": 0.1,
VERSION_FIELD: {
"revision": "r",
"filename": "f2",
"field": "v2",
},
},
],
{
"x": "v",
"y": "dvc_inferred_y_value",
"x_label": "v",
"y_label": "y",
},
id="multi_source_y_single_x",
),
],
)
def test_convert(