fixes import of eig from fmmax.utils to fmmax.eig (#41)

* fixes import of eig from fmmax.utils to fmmax.eig

* bumps up fmmax version to 1.0.0 in pyproject.toml

Author: julian-urban

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"]
+"__init__.py" = ["F401", "F821"]

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