Make it possible for the user to pass on solver parameters

src/pulse/__init__.py

@@ -28,6 +28,7 @@
     utils,
     viscoelasticity,
 )
+from .active_model import ActiveModel, Passive
 from .active_stress import ActiveStress
 from .boundary_conditions import BoundaryConditions, NeumannBC, RobinBC
 from .cardiac_model import CardiacModel
@@ -70,6 +71,8 @@
     "active_model",
     "active_stress",
     "ActiveStress",
+    "ActiveModel",
+    "Passive",
     "geometry",
     "Geometry",
     "NeumannBC",

src/pulse/problem.py

@@ -513,10 +513,35 @@ def reset(self):
         self.reset_states()
         self._init_forms()
 
-    def solve(self) -> bool:
+    def solve(
+        self,
+        rtol: float = 1e-10,
+        atol: float = 1e-6,
+        beta: float = 1.0,
+        update_old_states: bool = True,
+    ) -> bool:
+        """Solve nonlinear problem with Newton solver
+
+        Parameters
+        ----------
+        rtol : float, optional
+            Relative tolerance, by default 1e-10
+        atol : float, optional
+            Absolute tolerance, by default 1e-6
+        beta : float, optional
+            Damping parameter, by default 1.0
+        update_old_states : bool, optional
+            Whether to update old states before solving, by default True
+
+        Returns
+        -------
+        bool
+            True if converged, False otherwise
+        """
         logger.debug("Solving the system...")
-        self.update_old_states()
-        ret = self._solver.solve(rtol=1e-10, atol=1e-6)
+        if update_old_states:
+            self.update_old_states()
+        ret = self._solver.solve(rtol=rtol, atol=atol, beta=beta)
         self.update_fields()
 
         return ret