remove installation of redundant modules from flax PR tests
Attach a file by drag & drop or click to upload
