Ensure that the interface is notified to exit when an exception occurs, before the main thread is interrupted.

PiperOrigin-RevId: 947852128

Author: timblakely

ffn/inference/executor.py

@@ -381,6 +381,7 @@ def _dispatcher(self, data: jnp.ndarray, client_ids: Sequence[int]):
         ret = self._apply_fn(data)
       except Exception as e:  # pylint:disable=broad-except
         logging.exception(e)
+        self._interface.exit_request.set()
         # Terminate the whole program on failure.
         _thread.interrupt_main()  # pytype: disable=module-attr
         raise e