From 37cc7efe6ea947a860e7b0bdbb4e35202a447faf Mon Sep 17 00:00:00 2001 From: Rafael Pastrana Date: Sun, 26 Jan 2025 20:39:27 -0500 Subject: [PATCH] [Eq] Cleaned up old comments in `models.py` --- src/jax_fdm/equilibrium/models.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/src/jax_fdm/equilibrium/models.py b/src/jax_fdm/equilibrium/models.py index 92be386..8c2fc1c 100644 --- a/src/jax_fdm/equilibrium/models.py +++ b/src/jax_fdm/equilibrium/models.py @@ -215,9 +215,8 @@ def equilibrium_iterative_fn(params, xyz_init): """ This closure function avoids re-computing A and f_fixed throughout iterations because these two matrices remain constant during the fixed point search. - - TODO: Extract closure into function shared with the other nodes equilibrium function? """ + # TODO: Extract closure into function shared with the other nodes equilibrium function? A, f_fixed, xyz_fixed, load_state = params free = structure.indices_free @@ -242,8 +241,6 @@ def equilibrium_iterative_fn(params, xyz_init): "eta": eta, "verbose": verbose, "implicit": True} - # For jaxopt compatibility (as it does not support sparse matrices yet) - # "implicit": False if self.linearsolve_fn is spsolve else True} solver_kwargs = {"solver_config": solver_config, "f": equilibrium_iterative_fn, @@ -251,13 +248,10 @@ def equilibrium_iterative_fn(params, xyz_init): "x_init": xyz_init} if implicit_diff: - # xyz_new = fixed_point(solver, **solver_kwargs) return fixed_point(solver, **solver_kwargs) - # xyz_new = solver(**solver_kwargs) return solver(**solver_kwargs) - # return xyz_new # ---------------------------------------------------------------------- # Equilibrium state