From 01fed6f2ca17f7f43aa9299ae73da660713ec243 Mon Sep 17 00:00:00 2001 From: chertianser Date: Fri, 6 Dec 2024 13:27:25 -0500 Subject: [PATCH 1/2] stricter requirements due to constraints --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index a236acd..62d8469 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,4 +9,5 @@ typeguard==2.13.3 typing_extensions>=4.8.0 jaxtyping pytest>=7.4.3 - +numpy<2.0.0 +python>=3.9 From 8750475da45c1e0ee831bd1a3902c5e2b3569a5e Mon Sep 17 00:00:00 2001 From: chertianser Date: Fri, 6 Dec 2024 13:27:55 -0500 Subject: [PATCH 2/2] update deprecated function --- grad_dft/utils/chunk.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/grad_dft/utils/chunk.py b/grad_dft/utils/chunk.py index 1298754..f8060fc 100644 --- a/grad_dft/utils/chunk.py +++ b/grad_dft/utils/chunk.py @@ -19,7 +19,7 @@ from jax import numpy as jnp from jax.tree_util import tree_leaves, tree_map -from jax import linear_util as lu +from jax.extend import linear_util as lu from jax.api_util import argnums_partial from .types import Array