[gguf] RE_GGUF_SHARD_FILE named groups (#621)

Suggested by @coyotte508
huggingface · Apr 11, 2024 · e78bc67 · e78bc67
1 parent 5f213a9
commit e78bc67
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/packages/gguf/src/gguf.spec.ts b/packages/gguf/src/gguf.spec.ts
@@ -226,7 +226,7 @@ describe("gguf", () => {
 		const match = ggufPath.match(RE_GGUF_SHARD_FILE);
 
 		expect(RE_GGUF_SHARD_FILE.test(ggufPath)).toEqual(true);
-		expect(match?.[1]).toEqual("00003");
-		expect(match?.[2]).toEqual("00009");
+		expect(match?.groups?.shard).toEqual("00003");
+		expect(match?.groups?.total).toEqual("00009");
 	});
 });
diff --git a/packages/gguf/src/gguf.ts b/packages/gguf/src/gguf.ts
@@ -6,7 +6,7 @@ export { GGUFValueType, GGMLQuantizationType } from "./types";
 export { GGUF_QUANT_DESCRIPTIONS } from "./quant-descriptions";
 
 export const RE_GGUF_FILE = /\.gguf$/;
-export const RE_GGUF_SHARD_FILE = /-(\d{5})-of-(\d{5})\.gguf$/;
+export const RE_GGUF_SHARD_FILE = /-(?<shard>\d{5})-of-(?<total>\d{5})\.gguf$/;
 
 const isVersion = (version: number): version is Version => version === 1 || version === 2 || version === 3;