diff --git a/extension/src/pipeline/index.ts b/extension/src/pipeline/index.ts index ae019ad728..a92c75b1e1 100644 --- a/extension/src/pipeline/index.ts +++ b/extension/src/pipeline/index.ts @@ -17,6 +17,8 @@ import { getInput, getValidInput } from '../vscode/inputBox' import { Title } from '../vscode/title' import { quickPickOne, quickPickOneOrInput } from '../vscode/quickPick' import { pickFile } from '../vscode/resourcePicker' +import { Modal } from '../vscode/modal' +import { Response } from '../vscode/response' export enum ScriptCommand { JUPYTER = 'jupyter nbconvert --to notebook --inplace --execute', @@ -150,6 +152,16 @@ export class Pipeline extends DeferredDisposable { } private async addPipeline() { + const response = await Modal.showInformation( + 'To continue you must add a stage to the dvc.yaml file. Would you like to add a stage now?', + Response.YES, + Response.NO + ) + + if (response !== Response.YES) { + return + } + const stageName = await this.askForStageName() if (!stageName) { return