Skip to content

Commit

Permalink
Use xi instead of xi_epsed for evaluating alpha function for pl…
Browse files Browse the repository at this point in the history
…anar hsa
  • Loading branch information
mstoelzle committed Oct 27, 2023
1 parent 92a6bc5 commit f1c8ebc
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions src/jsrm/systems/planar_hsa.py
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,6 @@ 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"],
Expand Down Expand Up @@ -557,7 +556,7 @@ def dynamical_matrices_fn(
G = G_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()
alpha = alpha_lambda(*params_for_lambdify, *xi, *phi).squeeze()

# apply the strain basis
B = B_xi.T @ B @ B_xi
Expand Down

0 comments on commit f1c8ebc

Please sign in to comment.