[python] flush logging output before process end (#567)

deepjavalibrary · Mar 23, 2023 · 53c9718 · 53c9718
1 parent ea1b41c
commit 53c9718
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/engines/python/setup/djl_python_engine.py b/engines/python/setup/djl_python_engine.py
@@ -153,6 +153,9 @@ def main():
         logging.exception("Python engine process died")
     finally:
         logging.info(f"{pid} - Python process finished")
+        sys.stdout.flush()
+        sys.stderr.flush()
+        logging.shutdown()
         if sock_type == 'unix':
             if os.path.exists(sock_name):
                 os.remove(sock_name)