Debug browser tests

Author: coyotte508

packages/hub/src/utils/XetBlob.spec.ts

@@ -73,8 +73,8 @@ describe("XetBlob", () => {
 			}
 		).then((res) => res.arrayBuffer());
 
-		expect(new Uint8Array(xetDownload)).toEqual(new Uint8Array(bridgeDownload));
 		expect(xetDownload.byteLength).toBe(29929);
+		expect(new Uint8Array(xetDownload)).toEqual(new Uint8Array(bridgeDownload));
 		expect(xorbCount).toBe(2);
 	});
 
@@ -112,32 +112,29 @@ describe("XetBlob", () => {
 
 	// In github actions, this test doesn't work inside the browser, but it works locally
 	// inside both chrome and chromium browsers
-	// TODO: figure out why
-	if (typeof window === "undefined" && Math.random() === 10) {
-		it("should load correctly when loading far into a chunk range", async () => {
-			const blob = new XetBlob({
-				repo: {
-					type: "model",
-					name: "celinah/xet-experiments",
+	it("should load correctly when loading far into a chunk range", async () => {
+		const blob = new XetBlob({
+			repo: {
+				type: "model",
+				name: "celinah/xet-experiments",
+			},
+			hash: "7b3b6d07673a88cf467e67c1f7edef1a8c268cbf66e9dd9b0366322d4ab56d9b",
+			size: 5_234_139_343,
+		});
+
+		const xetDownload = await blob.slice(10_000_000, 10_100_000).arrayBuffer();
+		const bridgeDownload = await fetch(
+			"https://huggingface.co/celinah/xet-experiments/resolve/main/model5GB.safetensors",
+			{
+				headers: {
+					Range: "bytes=10000000-10099999",
 				},
-				hash: "7b3b6d07673a88cf467e67c1f7edef1a8c268cbf66e9dd9b0366322d4ab56d9b",
-				size: 5_234_139_343,
-			});
-
-			const xetDownload = await blob.slice(10_000_000, 10_100_000).arrayBuffer();
-			const bridgeDownload = await fetch(
-				"https://huggingface.co/celinah/xet-experiments/resolve/main/model5GB.safetensors",
-				{
-					headers: {
-						Range: "bytes=10000000-10099999",
-					},
-				}
-			).then((res) => res.arrayBuffer());
+			}
+		).then((res) => res.arrayBuffer());
 
-			console.log("xet", xetDownload.byteLength, "bridge", bridgeDownload.byteLength);
-			expect(new Uint8Array(xetDownload)).toEqual(new Uint8Array(bridgeDownload));
-		}, 30_000);
-	}
+		console.log("xet", xetDownload.byteLength, "bridge", bridgeDownload.byteLength);
+		expect(new Uint8Array(xetDownload)).toEqual(100_000);
+	}, 30_000);
 
 	it("should load text correctly when offset_into_range starts in a chunk further than the first", async () => {
 		const blob = new XetBlob({

packages/hub/src/utils/XetBlob.ts

@@ -279,6 +279,8 @@ export class XetBlob extends Blob {
 					const result = await reader.read();
 					done = result.done;
 
+					console.log("read", result.value?.length, "bytes");
+
 					if (!result.value) {
 						continue;
 					}
@@ -362,8 +364,10 @@ export class XetBlob extends Blob {
 						}
 
 						if (uncompressed.length) {
+							console.log("yield", uncompressed.length, "bytes", result.value.length, "total read", totalBytesRead);
 							totalBytesRead += uncompressed.length;
 							yield stored ? uncompressed.slice() : uncompressed;
+							console.log("yielded", uncompressed.length, "bytes", result.value.length, "total read", totalBytesRead);
 						}
 					}