Fix bug in specifying D

tud-phi · Nov 19, 2024 · e192e5d · e192e5d
1 parent fd05085
commit e192e5d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/simulate_planar_pcs.py b/examples/simulate_planar_pcs.py
@@ -36,10 +36,10 @@
     "G": 1e3 * jnp.ones((num_segments,)),  # Shear modulus [Pa]
 }
 params["D"] = 1e-3 * jnp.diag(
-    jnp.repeat(
+    (jnp.repeat(
         jnp.array([[1e0, 1e3, 1e3]]), num_segments, axis=0
-    ).flatten(),
-) * params["l"]
+    ) * params["l"][:, None]).flatten()
+)
 
 # activate all strains (i.e. bending, shear, and axial)
 strain_selector = jnp.ones((3 * num_segments,), dtype=bool)