Commit 147f0903 by vincent

fixed testcase for loading modelfile from external uri

parent 47a54b57
...@@ -60,10 +60,10 @@ describe('loadWeightMap', () => { ...@@ -60,10 +60,10 @@ describe('loadWeightMap', () => {
}) })
it('returns correct uris, given external path', () => { it('returns correct uris, given external path', () => {
const uri = 'https://example.com/path/to/modelfiles/'; const uri = 'https://example.com/path/to/modelfiles';
const result = getModelUris(uri, FAKE_DEFAULT_MODEL_NAME) const result = getModelUris(uri, FAKE_DEFAULT_MODEL_NAME)
expect(result.manifestUri).toEqual(uri + 'model-weights_manifest.json') expect(result.manifestUri).toEqual(`${uri}/${FAKE_DEFAULT_MODEL_NAME}-weights_manifest.json`)
expect(result.modelBaseUri).toEqual(uri) expect(result.modelBaseUri).toEqual(uri)
}) })
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment