remove docs

iterative · Jul 12, 2023 · 2c81c8b · 2c81c8b
1 parent 7b6022a
commit 2c81c8b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/dvc/schema.py b/dvc/schema.py
@@ -45,7 +45,7 @@
 LOCKFILE_STAGES_SCHEMA = {str: LOCK_FILE_STAGE_SCHEMA}
 LOCKFILE_SCHEMA = {
     Required("schema"): Equal("2.0", "invalid schema version"),
-    Required(STAGES): LOCKFILE_STAGES_SCHEMA,
+    STAGES: LOCKFILE_STAGES_SCHEMA,
 }
 
 OUT_PSTAGE_DETAILED_SCHEMA = {
@@ -93,12 +93,13 @@
 
 
 def stage_schema(data):
+    schema = STAGE_DEFINITION
     if isinstance(data, Mapping):
         if FOREACH_KWD in data:
-            return Schema(FOREACH_DO)(data)
-        if MATRIX_KWD in data:
-            return Schema(MATRIX_DO)(data)
-    return Schema(STAGE_DEFINITION)(data)
+            schema = FOREACH_DO
+        elif MATRIX_KWD in data:
+            schema = MATRIX_DO
+    return Schema(schema)(data)
 
 
 PLOT_DEFINITION = {

diff --git a/tests/func/parsing/test_matrix.py b/tests/func/parsing/test_matrix.py
@@ -23,7 +23,6 @@
     ],
 )
 def test_matrix_interpolated(tmp_dir, dvc, matrix):
-    """Testing a simple non-nested list as a foreach data"""
     (tmp_dir / "params.yaml").dump(MATRIX_DATA)
     resolver = DataResolver(dvc, tmp_dir.fs_path, {})
     data = {