chore: reorder dependencies

Author: XuehaiPan

.github/workflows/build.yml

@@ -42,9 +42,11 @@ concurrency:
   cancel-in-progress: ${{ github.event_name == 'pull_request' }}
 
 env:
+  OPTREE_CXX_WERROR: "OFF"
+  _GLIBCXX_USE_CXX11_ABI: "1"
+  PYTHONUNBUFFERED: "1"
   PYTHON_TAG: "py3" # to be updated
   PYTHON_VERSION: "3" # to be updated
-  _GLIBCXX_USE_CXX11_ABI: "1"
   COLUMNS: "100"
   FORCE_COLOR: "1"
   CLICOLOR_FORCE: "1"

docs/requirements.txt

@@ -3,14 +3,14 @@
 sphinx
 sphinx-autoapi
 sphinx-autobuild
+sphinx-autodoc-typehints
 sphinx-copybutton
 sphinx-rtd-theme
 sphinxcontrib-bibtex
-sphinx-autodoc-typehints
 docutils
 
 --extra-index-url https://download.pytorch.org/whl/cpu
 
-jax[cpu] >= 0.4.6
+jax[cpu]
 numpy
 torch

pyproject.toml

@@ -58,19 +58,20 @@ jax = ["jax"]
 numpy = ["numpy"]
 torch = ["torch"]
 lint = [
-    "ruff",
-    "pylint[spelling]",
-    "mypy",
+    "cpplint",
     "doc8",
+    "mypy",
+    "pre-commit",
     "pyenchant",
+    "pylint[spelling]",
+    "ruff",
     "xdoctest",
-    "cpplint",
-    "pre-commit",
 ]
 test = [
     "pytest",
     "pytest-cov",
     "covdefaults",
+    "xdoctest",
     "rich",
     # Pin to minimum for compatibility test
     "typing-extensions == 4.6.0; python_version < '3.13' and platform_system == 'Linux'",
@@ -84,10 +85,10 @@ docs = [
     "sphinx",
     "sphinx-autoapi",
     "sphinx-autobuild",
+    "sphinx-autodoc-typehints",
     "sphinx-copybutton",
     "sphinx-rtd-theme",
     "sphinxcontrib-bibtex",
-    "sphinx-autodoc-typehints",
     "docutils",
     "jax[cpu]",
     "numpy",

tests/concurrent/test_subinterpreters.py

@@ -30,7 +30,7 @@
     pytest.skip('Test for CPython 3.14+ only', allow_module_level=True)
 
 
-from concurrent.futures import InterpreterPoolExecutor
+from concurrent.futures import InterpreterPoolExecutor, as_completed
 
 
 if Py_GIL_DISABLED and not Py_DEBUG:
@@ -51,3 +51,16 @@ def run(func, /, *args, **kwargs):
     if exception is not None:
         raise exception
     return future.result()
+
+
+def concurrent_run(func, /, *args, **kwargs):
+    futures = [EXECUTOR.submit(func, *args, **kwargs) for _ in range(NUM_FUTURES)]
+    future2index = {future: i for i, future in enumerate(futures)}
+    completed_futures = sorted(as_completed(futures), key=future2index.get)
+    first_exception = next(filter(None, (future.exception() for future in completed_futures)), None)
+    if first_exception is not None:
+        raise first_exception
+    return [future.result() for future in completed_futures]
+
+
+run(object)  # warm-up

tests/concurrent/test_threading.py

@@ -55,6 +55,14 @@
 atexit.register(EXECUTOR.shutdown)
 
 
+def run(func, /, *args, **kwargs):
+    future = EXECUTOR.submit(func, *args, **kwargs)
+    exception = future.exception()
+    if exception is not None:
+        raise exception
+    return future.result()
+
+
 def concurrent_run(func, /, *args, **kwargs):
     futures = [EXECUTOR.submit(func, *args, **kwargs) for _ in range(NUM_FUTURES)]
     future2index = {future: i for i, future in enumerate(futures)}
@@ -65,7 +73,7 @@ def concurrent_run(func, /, *args, **kwargs):
     return [future.result() for future in completed_futures]
 
 
-concurrent_run(object)  # warm-up
+run(object)  # warm-up
 
 
 @parametrize(

tests/requirements.txt

@@ -3,16 +3,19 @@
 pytest
 pytest-cov
 covdefaults
-rich
-ruff
-pylint[spelling]
-mypy
-doc8
-pyenchant
 xdoctest
+rich
+
 cpplint
+doc8
+mypy
 pre-commit
+pyenchant
+pylint[spelling]
+ruff
+
+--extra-index-url https://download.pytorch.org/whl/cpu
 
-jax[cpu] >= 0.4.6
+jax[cpu]
 numpy
 torch