Skip to content

Commit 19a3be2

Browse files
committed
test: fix types
1 parent e1db162 commit 19a3be2

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

test/modelDependent/stableCode/asyncContextLoad.test.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ describe("stableCode", () => {
5151
}
5252

5353
expect(loopIterationsBeforeUnload).toBeGreaterThanOrEqual(2);
54-
expect(disposePromise).resolves.toBeUndefined();
54+
await expect(disposePromise).resolves.toBeUndefined();
5555
});
5656
});
5757
});

test/modelDependent/stableCode/asyncModelLoad.test.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ describe("stableCode", () => {
4848
}
4949

5050
expect(loopIterationsBeforeUnload).toBeGreaterThanOrEqual(2);
51-
expect(disposePromise).resolves.toBeUndefined();
51+
await expect(disposePromise).resolves.toBeUndefined();
5252
});
5353

5454
test("load progress emitted", {timeout: 1000 * 60 * 60 * 2}, async () => {
@@ -62,7 +62,7 @@ describe("stableCode", () => {
6262
const modelPromise = llama.loadModel({
6363
modelPath,
6464
onLoadProgress(loadPercentage: number) {
65-
if (logProgresses.length === 0 || loadPercentage - logProgresses[logProgresses.length - 1] >= 0.1 || loadPercentage === 1)
65+
if (logProgresses.length === 0 || loadPercentage - logProgresses.at(-1)! >= 0.1 || loadPercentage === 1)
6666
logProgresses.push(loadPercentage);
6767
}
6868
});
@@ -106,7 +106,7 @@ describe("stableCode", () => {
106106
modelPath,
107107
loadSignal: loadController.signal,
108108
onLoadProgress(loadPercentage: number) {
109-
if (logProgresses.length === 0 || loadPercentage - logProgresses[logProgresses.length - 1] >= 0.1 || loadPercentage === 1)
109+
if (logProgresses.length === 0 || loadPercentage - logProgresses.at(-1)! >= 0.1 || loadPercentage === 1)
110110
logProgresses.push(loadPercentage);
111111

112112
if (loadPercentage >= 0.2 && !loadController.signal.aborted) {

0 commit comments

Comments
 (0)