tighten schema for plots; require key name

iterative · Jan 12, 2024 · 3a82fba · 3a82fba
1 parent cfeb6bc
commit 3a82fba
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/dvc/repo/plots/__init__.py b/dvc/repo/plots/__init__.py
@@ -486,7 +486,8 @@ def _collect_pipeline_files(repo, targets: List[str], props, onerror=None):
         for elem in plots_def:
             if isinstance(elem, str):
                 dvcfile_defs_dict[elem] = None
-            elif elem:
+            else:
+                assert elem
                 k, v = next(iter(elem.items()))
                 dvcfile_defs_dict[k] = v
 

diff --git a/dvc/schema.py b/dvc/schema.py
@@ -112,7 +112,7 @@ def validator(data):
     Output.PARAM_PLOT_TITLE: str,
     Output.PARAM_PLOT_TEMPLATE: str,
 }
-SINGLE_PLOT_SCHEMA = {str: vol.Any(PLOT_DEFINITION, None)}
+SINGLE_PLOT_SCHEMA = {vol.Required(str): vol.Any(PLOT_DEFINITION, None)}
 ARTIFACTS = "artifacts"
 SINGLE_ARTIFACT_SCHEMA = vol.Schema({str: ARTIFACT_SCHEMA})
 FOREACH_IN = {