From 5f0501bac8015a48fc6cf78a6487a714df2b6553 Mon Sep 17 00:00:00 2001
From: Ross Hemsley <rhemsley@google.com>
Date: Mon, 21 Nov 2022 01:57:17 -0800
Subject: [PATCH] Use auto instead of `/proc/cpuinfo`

We can't always rely on this working for all platforms.
Fixes #450

PiperOrigin-RevId: 489921823
---
 test.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test.sh b/test.sh
index 5e43b0090..4e30f5ca8 100755
--- a/test.sh
+++ b/test.sh
@@ -70,11 +70,11 @@ pytype `find optax/_src/ examples -name '*.py' | xargs` -k -d import-error
 # Run tests using pytest.
 # Change directory to avoid importing the package from repo root.
 mkdir _testing && cd _testing
-python -m pytest -n "$(grep -c ^processor /proc/cpuinfo)" --pyargs optax
+python -m pytest -n auto --pyargs optax
 cd ..
 
 cd examples
-python -m pytest -n "$(grep -c ^processor /proc/cpuinfo)" .
+python -m pytest -n auto .
 cd ..
 
 # Build Sphinx docs.