From 56d5f3627b26b8ca13988cafcc9fa189d09441a9 Mon Sep 17 00:00:00 2001
From: Dario Coscia <93731561+dario-coscia@users.noreply.github.com>
Date: Wed, 3 Apr 2024 18:29:42 +0200
Subject: [PATCH] Update parametric_poisson.py (#276)

---
 examples/problems/parametric_poisson.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/problems/parametric_poisson.py b/examples/problems/parametric_poisson.py
index c6581cb..58867d5 100644
--- 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)),
-    }
\ No newline at end of file
+    }