minor changes

2022-02-11 17:24:12 +01:00
parent 1483746b45
commit 1812ddb8d9
5 changed files with 12 additions and 13 deletions
--- a/examples/problems/burgers.py
+++ b/examples/problems/burgers.py
@@ -15,7 +15,8 @@ class Burgers1D(TimeDependentProblem, SpatialProblem):

    def burger_equation(input_, output_):
        grad_u = grad(output_['u'], input_)
-        grad_x, grad_t = grad(output_['u'], input_).T
+        grad_x = grad_u['x']
+        grad_t = grad_u['t']
        gradgrad_u_x = grad(grad_u['x'], input_)
        return (
            grad_u['t'] + output_['u']*grad_u['x'] -