Update parametric_poisson.py (#276)

2024-04-03 18:29:42 +02:00
parent 14dfa62402
commit 56d5f3627b
1 changed files with 2 additions and 2 deletions
--- a/examples/problems/parametric_poisson.py
+++ b/examples/problems/parametric_poisson.py
@@ -33,7 +33,7 @@ class ParametricPoisson(SpatialProblem, ParametricProblem):
        force_term = torch.exp(
                - 2*(input_.extract(['x']) - input_.extract(['mu1']))**2
                - 2*(input_.extract(['y']) - input_.extract(['mu2']))**2)
-        return laplacian(output_.extract(['u']), input_) - force_term
+        return laplacian(output_.extract(['u']), input_, d=['x','y']) - force_term

    # problem condition statement
    conditions = {
@@ -52,4 +52,4 @@ class ParametricPoisson(SpatialProblem, ParametricProblem):
        'D': Condition(
            location=CartesianDomain({'x': [-1, 1], 'y': [-1, 1], 'mu1': [-1, 1], 'mu2': [-1, 1]}),
            equation=Equation(laplace_equation)),
-    }
+    }