preliminary modifications for N-S

examples/problems/parametric_poisson.py

@@ -14,13 +14,13 @@ class ParametricPoisson(SpatialProblem, ParametricProblem):
 
     def laplace_equation(input_, output_):
         force_term = torch.exp(
-                - 2*(input_['x'] - input_['mu1'])**2 - 2*(input_['y'] -
-                                                          input_['mu2'])**2)
-        return nabla(output_['u'], input_) - force_term
+                - 2*(input_.extract(['x']) - input_.extract(['mu1']))**2 - 2*(input_.extract(['y']) -
+                                                          input_.extract(['mu2']))**2)
+        return nabla(output_.extract(['u']), input_) - force_term
 
     def nil_dirichlet(input_, output_):
         value = 0.0
-        return output_['u'] - value
+        return output_.extract(['u']) - value
 
     conditions = {
         'gamma1': Condition(