diff --git a/equinox/_jit.py b/equinox/_jit.py index 9857cd30..8849b794 100644 --- a/equinox/_jit.py +++ b/equinox/_jit.py @@ -161,6 +161,10 @@ def write(self, data: str): if "_EquinoxRuntimeError" not in data: self.stderr.write(data) + # Needed for the PyCharm debugger, see #827. + def flush(self): + self.stderr.flush() + class _JitWrapper(Module): fn: str # this attribute exists solely to give a nice repr