Fix cmd definition union issue for VS Code

iterative · Jul 7, 2023 · b44278c · b44278c
1 parent 2296394
commit b44278c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/gen.py b/gen.py
@@ -17,6 +17,10 @@
 Template = PlotTemplateName | FilePath
 
 
+class Cmds(BaseModel):
+    __root__: list[str]
+
+
 class OutFlags(BaseModel):
     cache: bool | None = Field(True, description="Cache output by DVC")
     persist: bool | None = Field(
@@ -247,7 +251,7 @@ class Stage(BaseModel):
     A named stage of a pipeline.
     """
 
-    cmd: str | list[str] = Field(..., description=CMD_DESC)
+    cmd: str | Cmds = Field(..., description=CMD_DESC)
     wdir: str | None = Field(
         None,
         description="Working directory for the cmd, relative to `dvc.yaml`",

diff --git a/schema.json b/schema.json
@@ -92,6 +92,13 @@
         ]
       }
     },
+    "Cmds": {
+      "title": "Cmds",
+      "type": "array",
+      "items": {
+        "type": "string"
+      }
+    },
     "DepModel": {
       "title": "DepModel",
       "description": "Path to a dependency (input) file or directory for the stage.",
@@ -367,10 +374,7 @@
               "type": "string"
             },
             {
-              "type": "array",
-              "items": {
-                "type": "string"
-              }
+              "$ref": "#/definitions/Cmds"
             }
           ]
         },