From 4c6e9c232ced9d7d599b08ae85549aa2ee866356 Mon Sep 17 00:00:00 2001 From: Julian Urban Date: Tue, 18 Mar 2025 10:54:52 -0400 Subject: [PATCH 1/2] fixes import of eig from fmmax.utils to fmmax.eig --- thermox/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/thermox/utils.py b/thermox/utils.py index c08e8a6..c47e038 100644 --- a/thermox/utils.py +++ b/thermox/utils.py @@ -1,7 +1,7 @@ from typing import NamedTuple, Tuple from jax import numpy as jnp from jax import Array -from fmmax.utils import ( +from fmmax.eig import ( eig, ) # differentiable and jit-able eigendecomposition, not yet available in jax, see https://github.com/google/jax/issues/2748 From 16129bac64bf639eaa2451d207ee267a32516e86 Mon Sep 17 00:00:00 2001 From: Julian Urban Date: Tue, 18 Mar 2025 15:19:29 -0400 Subject: [PATCH 2/2] bumps up fmmax version to 1.0.0 in pyproject.toml --- pyproject.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 8fc1d9b..38bb04d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -15,7 +15,7 @@ classifiers = [ "Topic :: Scientific/Engineering :: Mathematics", "License :: OSI Approved :: Apache Software License", ] -dependencies = ["jax>=0.4.0", "jaxlib>=0.4.0", "fmmax>=0.8.0"] +dependencies = ["jax>=0.4.0", "jaxlib>=0.4.0", "fmmax>=1.0.0"] [project.optional-dependencies] test = ["pre-commit", "pytest-cov", "ruff", "optax", "mypy"] @@ -25,4 +25,4 @@ packages = ["thermox"] [tool.ruff] [tool.ruff.lint.per-file-ignores] -"__init__.py" = ["F401", "F821"] \ No newline at end of file +"__init__.py" = ["F401", "F821"]