def find_vega(repo, plots_data, target): found = dpath.util.search(plots_data, ["*", "*", target]) from dvc.render.vega import VegaRenderer if found and VegaRenderer.matches(found): return VegaRenderer(found, repo.plots.templates).get_vega() return ""
def match_renderers(plots_data, templates): from dvc.render.image import ImageRenderer from dvc.render.vega import VegaRenderer renderers = [] for g in group_by_filename(plots_data): if VegaRenderer.matches(g): renderers.append(VegaRenderer(g, templates)) if ImageRenderer.matches(g): renderers.append(ImageRenderer(g)) return renderers
def test_matches(extension, matches): filename = "file" + extension data = { "HEAD": { "data": { filename: {} } }, "v1": { "data": { filename: {} } }, } assert VegaRenderer.matches(data) == matches
def find_vega(repo, plots_data, target): # TODO same as group_by_filename grouped = group_by_filename(plots_data) found = None for plot_group in grouped: files = get_files(plot_group) assert len(files) == 1 file = files.pop() if file == target: found = plot_group break from dvc.render.vega import VegaRenderer if found and VegaRenderer.matches(found): return VegaRenderer(found, repo.plots.templates).get_vega() return ""