Skip to content

Commit 1d367d0

Browse files
committed
drop terrible idea of holding all data as strings
1 parent b50b671 commit 1d367d0

File tree

3 files changed

+16
-14
lines changed

3 files changed

+16
-14
lines changed

dvc/commands/plots.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def run(self) -> int: # noqa: C901, PLR0911, PLR0912
135135
if self.args.show_vega:
136136
renderer = first(filter(lambda r: r.TYPE == "vega", renderers))
137137
if renderer:
138-
ui.write_json(renderer.get_filled_template(as_string=False))
138+
ui.write_json(renderer.get_filled_template())
139139
return 0
140140

141141
output_file: Path = (Path.cwd() / out).resolve() / "index.html"

dvc/render/convert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def to_json(renderer, split: bool = False) -> List[Dict]:
2626
if split:
2727
content, split_content = renderer.get_partial_filled_template()
2828
else:
29-
content = renderer.get_filled_template(as_string=False)
29+
content = renderer.get_filled_template()
3030
split_content = {}
3131

3232
return [

tests/integration/plots/test_plots.py

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -114,16 +114,20 @@ def verify_vega(
114114
assert j[0]["type"] == "vega"
115115
assert set(j[0]["revisions"]) == set(versions)
116116

117-
assert json_result[0]["content"]["data"]["values"] == json.loads(
118-
split_json_result[0]["anchor_definitions"]["<DVC_METRIC_DATA>"]
117+
assert (
118+
json_result[0]["content"]["data"]["values"]
119+
== split_json_result[0]["anchor_definitions"]["<DVC_METRIC_DATA>"]
119120
)
121+
120122
assert set(versions) == set(json_result[0]["revisions"])
121123

122124
assert json_result[0]["content"]["data"]["values"]
123125
assert html_result["data"]["values"]
124-
assert "<DVC_METRIC_DATA>" in split_json_result[0]["content"]
125-
assert "<DVC_METRIC_X_LABEL>" in split_json_result[0]["content"]
126-
assert "<DVC_METRIC_Y_LABEL>" in split_json_result[0]["content"]
126+
127+
content_str = json.dumps(split_json_result[0]["content"])
128+
assert "<DVC_METRIC_DATA>" in content_str
129+
assert "<DVC_METRIC_X_LABEL>" in content_str
130+
assert "<DVC_METRIC_Y_LABEL>" in content_str
127131

128132
def _assert_templates_equal(
129133
html_template, filled_template, split_template, title, x_label, y_label
@@ -133,7 +137,7 @@ def _assert_templates_equal(
133137
tmp1 = deepcopy(html_template)
134138
tmp2 = deepcopy(filled_template)
135139
tmp3 = json.loads(
136-
split_template[:]
140+
json.dumps(split_template)
137141
.replace('"<DVC_METRIC_COLUMN_WIDTH>"', "300")
138142
.replace('"<DVC_METRIC_PLOT_HEIGHT>"', "300")
139143
.replace('"<DVC_METRIC_PLOT_WIDTH>"', "300")
@@ -486,11 +490,9 @@ def test_repo_with_config_plots(tmp_dir, capsys, repo_with_config_plots):
486490

487491
assert html_result["linear_train_vs_test"]["data"]["values"] == ble
488492
assert (
489-
json.loads(
490-
split_json_result["data"]["linear_train_vs_test"][0]["anchor_definitions"][
491-
"<DVC_METRIC_DATA>"
492-
]
493-
)
493+
split_json_result["data"]["linear_train_vs_test"][0]["anchor_definitions"][
494+
"<DVC_METRIC_DATA>"
495+
]
494496
== ble
495497
)
496498

@@ -501,7 +503,7 @@ def test_repo_with_dvclive_plots(tmp_dir, capsys, repo_with_dvclive_plots):
501503

502504
for s in ("show", "diff"):
503505
_, json_result, split_json_result = call(capsys, subcommand=s)
504-
expected_result = {
506+
expected_result: Dict[str, Dict[str, list[str]]] = {
505507
"data": {
506508
"dvclive/plots/metrics/metric.tsv": [],
507509
},

0 commit comments

Comments
 (0)