Add _FilteredStderr.flush.

Fixes #827.
patrick-kidger · Sep 3, 2024 · 14e2c49 · 14e2c49
1 parent 3cdeef8
commit 14e2c49
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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