From 477a2029e1b8884d3e63c478854cc332599c4d7f Mon Sep 17 00:00:00 2001
From: Owen Lockwood <42878312+lockwo@users.noreply.github.com>
Date: Thu, 15 Aug 2024 14:10:30 -0700
Subject: [PATCH 1/3] sanic's the name, speeds the game

---
 test/__main__.py      | 2 +-
 test/requirements.txt | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/test/__main__.py b/test/__main__.py
index 474e1b47..2b9f7f52 100644
--- a/test/__main__.py
+++ b/test/__main__.py
@@ -12,7 +12,7 @@
 running_out = 0
 for file in here.iterdir():
     if file.is_file() and file.name.startswith("test"):
-        cmd = f"pytest {file} " + flags
+        cmd = f"pytest -n auto {file} " + flags
         out = subprocess.run(cmd, shell=True).returncode
         running_out = max(running_out, out)
 sys.exit(running_out)
diff --git a/test/requirements.txt b/test/requirements.txt
index 9de88eb6..723e3188 100644
--- a/test/requirements.txt
+++ b/test/requirements.txt
@@ -2,5 +2,6 @@ beartype
 jaxlib
 optax
 pytest
+pytest-xdist
 scipy
 tqdm

From 11b2baa74acea36fa527075c9a36889d810ac140 Mon Sep 17 00:00:00 2001
From: Owen Lockwood <42878312+lockwo@users.noreply.github.com>
Date: Thu, 15 Aug 2024 15:02:16 -0700
Subject: [PATCH 2/3] oom (reducing 4 -> 3)

---
 test/__main__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/__main__.py b/test/__main__.py
index 2b9f7f52..abadcb93 100644
--- a/test/__main__.py
+++ b/test/__main__.py
@@ -12,7 +12,7 @@
 running_out = 0
 for file in here.iterdir():
     if file.is_file() and file.name.startswith("test"):
-        cmd = f"pytest -n auto {file} " + flags
+        cmd = f"pytest -n 3 {file} " + flags
         out = subprocess.run(cmd, shell=True).returncode
         running_out = max(running_out, out)
 sys.exit(running_out)

From 9a138228e5f8c193c54ba054749dc5a53319d0a8 Mon Sep 17 00:00:00 2001
From: Owen Lockwood <42878312+lockwo@users.noreply.github.com>
Date: Thu, 15 Aug 2024 15:49:04 -0700
Subject: [PATCH 3/3] just 2?!

---
 test/__main__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/__main__.py b/test/__main__.py
index abadcb93..ba38d4e6 100644
--- a/test/__main__.py
+++ b/test/__main__.py
@@ -12,7 +12,7 @@
 running_out = 0
 for file in here.iterdir():
     if file.is_file() and file.name.startswith("test"):
-        cmd = f"pytest -n 3 {file} " + flags
+        cmd = f"pytest -n 2 {file} " + flags
         out = subprocess.run(cmd, shell=True).returncode
         running_out = max(running_out, out)
 sys.exit(running_out)