Skip to content

Commit d56f717

Browse files
committed
Use MultiCancellationToken in model evaluator
1 parent 8bb0085 commit d56f717

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

extensions/ql-vscode/src/model-editor/model-evaluator.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ import type { CancellationToken } from "vscode";
1818
import { CancellationTokenSource } from "vscode";
1919
import type { QlPackDetails } from "../variant-analysis/ql-pack-details";
2020
import type { App } from "../common/app";
21+
import { MultiCancellationToken } from "../common/vscode/multi-cancellation-token";
2122
import { ModelAlertsView } from "./model-alerts/model-alerts-view";
2223

2324
export class ModelEvaluator extends DisposableObject {
@@ -68,11 +69,11 @@ export class ModelEvaluator extends DisposableObject {
6869

6970
// Submit variant analysis and monitor progress
7071
return withProgress(
71-
(progress) =>
72+
(progress, token) =>
7273
this.runVariantAnalysis(
7374
qlPack,
7475
progress,
75-
this.cancellationSource.token,
76+
new MultiCancellationToken(token, this.cancellationSource.token),
7677
),
7778
{
7879
title: "Run model evaluation",

0 commit comments

Comments
 (0)