Use xi and not xi_epsed for evaluating stiffness

tud-phi · Oct 27, 2023 · 92a6bc5 · 92a6bc5
1 parent fc37e75
commit 92a6bc5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/jsrm/systems/planar_hsa.py b/src/jsrm/systems/planar_hsa.py
@@ -8,9 +8,7 @@
 from .utils import (
     concatenate_params_syms,
     compute_strain_basis,
-    compute_planar_stiffness_matrix,
 )
-from jsrm.math_utils import blk_diag
 
 
 def factory(
@@ -154,6 +152,7 @@ def select_params_for_lambdify_fn(params: Dict[str, Array]) -> List[Array]:
         params_syms_cat + sym_exps["state_syms"]["xi"], sym_exps["exps"]["K"], "jax"
     )
     D_lambda = sp.lambdify(params_syms_cat, sym_exps["exps"]["D"], "jax")
+    print("alpha exp:\n", sym_exps["exps"]["alpha"])
     alpha_lambda = sp.lambdify(
         params_syms_cat + sym_exps["state_syms"]["xi"] + sym_exps["state_syms"]["phi"],
         sym_exps["exps"]["alpha"],
@@ -556,7 +555,7 @@ def dynamical_matrices_fn(
         B = B_lambda(*params_for_lambdify, *xi_epsed)
         C_xi = C_lambda(*params_for_lambdify, *xi_epsed, *xi_d)
         G = G_lambda(*params_for_lambdify, *xi_epsed).squeeze()
-        K = K_lambda(*params_for_lambdify, *xi_epsed).squeeze()
+        K = K_lambda(*params_for_lambdify, *xi).squeeze()
         D = D_lambda(*params_for_lambdify)
         alpha = alpha_lambda(*params_for_lambdify, *xi_epsed, *phi).squeeze()