Compatibility with numpy 2

.github/workflows/release.yml

@@ -10,7 +10,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Release
-        uses: patrick-kidger/action_update_python_project@v2
+        uses: patrick-kidger/action_update_python_project@v6
         with:
             python-version: "3.11"
             test-script: |
@@ -20,7 +20,3 @@ jobs:
             pypi-token: ${{ secrets.pypi_token }}
             github-user: patrick-kidger
             github-token: ${{ github.token }}
-            email-user: ${{ secrets.email_user }}
-            email-token: ${{ secrets.email_token }}
-            email-server: ${{ secrets.email_server }}
-            email-target: ${{ secrets.email_target }}

.github/workflows/run_tests.yml

@@ -7,7 +7,7 @@ jobs:
   run-tests:
     strategy:
       matrix:
-        python-version: [ 3.9 ]
+        python-version: [ "3.10", "3.12" ]
         os: [ ubuntu-latest ]
       fail-fast: false
     runs-on: ${{ matrix.os }}

.pre-commit-config.yaml

@@ -1,23 +1,14 @@
 repos:
-  - repo: https://github.com/ambv/black
-    rev: 22.3.0
-    hooks:
-    - id: black
-  - repo: https://github.com/charliermarsh/ruff-pre-commit
-    rev: 'v0.0.255'
-    hooks:
-      - id: ruff
-        args: ["--fix"]
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.2.2
+    hooks:
+      - id: ruff-format  # formatter
+        types_or: [ python, pyi, jupyter ]
+      - id: ruff  # linter
+        types_or: [ python, pyi, jupyter ]
+        args: [ --fix ]
   - repo: https://github.com/RobertCraigie/pyright-python
-    rev: v1.1.315
+    rev: v1.1.350
     hooks:
     - id: pyright
-      additional_dependencies: ["equinox", "jax", "sympy"]
-  - repo: https://github.com/nbQA-dev/nbQA
-    rev: 1.6.3
-    hooks:
-    - id: nbqa-black
-      additional_dependencies: [ipython==8.12, black]
-    - id: nbqa-ruff 
-      args: ["--ignore=I001"]
-      additional_dependencies: [ipython==8.12, ruff]
+      additional_dependencies: [equinox, jax, sympy]

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sympy2jax"
-version = "0.0.5"
+version = "0.0.6"
 description = "Turn SymPy expressions into trainable JAX expressions."
 readme = "README.md"
 requires-python ="~=3.9"

sympy2jax/sympy_module.py

@@ -129,7 +129,7 @@ class _Symbol(_AbstractNode):
     _name: str
 
     def __init__(self, expr: sympy.Expr):
-        self._name = expr.name  # pyright: ignore
+        self._name = str(expr.name)  # pyright: ignore
 
     def __call__(self, memodict: dict):
         try:

tests/test_symbolic_module.py

@@ -57,8 +57,8 @@ def assert_sympy_allclose(x, y):
 
 def test_example():
     x_sym = sympy.symbols("x_sym")
-    cosx = 1.0 * sympy.cos(x_sym)
-    sinx = 2.0 * sympy.sin(x_sym)
+    cosx = 1.0 * sympy.cos(x_sym)  # pyright: ignore[reportOperatorIssue]
+    sinx = 2.0 * sympy.sin(x_sym)  # pyright: ignore[reportOperatorIssue]
     mod = sympy2jax.SymbolicModule([cosx, sinx])
 
     x = jax.numpy.zeros(3)