search

rlouf / mcx

Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.
https://rlouf.github.io/mcx
Apache License 2.0
325 stars 17 forks source link

Make the tests run in parallel #25

Closed rlouf closed 4 years ago

rlouf commented 4 years ago

Testing the distribution already takes some time (~ 2 minutes) so we would already benefit from running the tests in parallel. I added pytest-xdist to the workflow that runs the tests and set up the number of threads to use to 4.