Update _base.py

adtzlr · Nov 9, 2024 · b70320e · b70320e
1 parent 3358876
commit b70320e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/felupe/constitution/_base.py b/src/felupe/constitution/_base.py
@@ -460,12 +460,12 @@ def constitutive_material(Material, name=None):
         ...
         ...     def gradient(self, x):
         ...         F, statevars = x[0], x[-1]
-        ...         dWdF = self.kwargs["a"] * fem.math.identity(F)
+        ...         dWdF = self.kwargs["a"] * np.eye(3).reshape(3, 3, 1, 1)
         ...         return [dWdF, statevars]
         ...
         ...     def hessian(self, x, **kwargs):
         ...         F, statevars = x[0], x[-1]
-        ...         d2WdFdF = self.kwargs["a"] * np.zeros((3, 3, 3, 3, *F.shape[2:]))
+        ...         d2WdFdF = self.kwargs["a"] * np.zeros((3, 3, 3, 3, 1, 1))
         ...         return [d2WdFdF]
         >>>
         >>> MyMaterial = fem.constitutive_material(MyMaterialFormulation)