TomAugspurger · Oct 16, 2026 · Oct 16, 2026 · Oct 16, 2026 · Oct 16, 2026 · Oct 16, 2026
@@ -539,12 +539,6 @@ def _reset(
         existing_executor_options = self.config.get("executor_options", {})
         if not isinstance(existing_executor_options, dict):
             existing_executor_options = {}
-        existing_quent_context = existing_executor_options.get("quent_context")
-        if (
-            "quent_context" in executor_options
-            and executor_options["quent_context"] != existing_quent_context
-        ):
-            raise ValueError("quent_context cannot be changed during reset")
 
         _disallowed_exec = {"num_py_executors"} & executor_options.keys()
         if _disallowed_exec:
@@ -589,6 +583,12 @@ def __exit__(self, *_: object) -> None:
         """Exit the context manager, calling :meth:`shutdown`."""
         self.shutdown()
 
+    def _set_quent(self, quent_context: QuentConfig | None) -> QuentConfig | None:
+        """Replace this engine's Quent runtimes and return the effective config."""
-        """Replace this engine's Quent runtimes and return the effective config."""
+        """
+        Replace this engine's Quent runtimes.
+
+        Closes any running Quent runtimes, then starts new ones for
+        ``quent_context`` unless it is ``None``.
+
+        Parameters
+        ----------
+        quent_context
+            Quent configuration to start, or ``None`` to only close.
+
+        Returns
+        -------
+        The effective Quent configuration, which may differ from
+        ``quent_context``, or ``None`` if Quent is disabled.
+        """
-        """Replace this engine's Quent runtimes and return the effective config."""
+        """
+        Replace this engine's Quent runtimes.
+
+        Closes any running Quent runtimes, then starts new ones for
+        ``quent_context`` unless it is ``None``.
+
+        Parameters
+        ----------
+        quent_context
+            Quent configuration to start, or ``None`` to only close.
+
+        Returns
+        -------
+        The effective Quent configuration, which may differ from
+        ``quent_context``, or ``None`` if Quent is disabled.
+        """
+        raise NotImplementedError(  # pragma: no cover
+            f"{type(self).__name__} does not implement _set_quent"
+        )
+
     def _run(self, func: Callable[..., T], *args: Any, **kwargs: Any) -> list[T]:
         """
         Execute a function on all ranks.