python312Packages.numpyro: reduce number of pytest parallel workers to reduce RAM usage (#372111)

2025-01-09 09:13:05 -05:00
parent f8dddc4a95 548207baea
commit efc8686dfc
1 changed files with 3 additions and 0 deletions
@@ -64,6 +64,9 @@ buildPythonPackage rec {
  pythonImportsCheck = [ "numpyro" ];

  pytestFlagsArray = [
+    # Tests memory consumption grows significantly with the number of parallel processes (reaches ~200GB with 80 jobs)
+    "--maxprocesses=8"
+
    # A few tests fail with:
    # UserWarning: There are not enough devices to run parallel chains: expected 2 but got 1.
    # Chains will be drawn sequentially. If you are running MCMC in CPU, consider using `numpyro.set_host_device_count(2)` at the beginning of your program.