Getting compile errors when running `npm test` using gpu package

When I install `layout-gpu` to use GPU accelerated algorithms, they run fine in my react app, but when I run a test command I get this error `Error: Cannot find module '@antv/layout@alpha'`. 

Which is quite confusing? I can' t see a `layout@alpha` package anywhere. Also fine it works in react dev mode but not when the app runs `npm i` to test. Note I am using vite and typescript.