change print to sys.stdout.write and flush in benchmark_runner.py (#2155

)

* change print to sys.stdout.write and flush in benchmark_runner.py

* use print with flush arg instead of sys.stdout.write and flush

---------

Co-authored-by: Russell Liu <russell.liu@uwaterloo.ca>

benchmark/benchmark_runner.py

@@ -242,7 +242,7 @@ def run_kuzu(serialized_graph_path):
         process = subprocess.Popen(
             tuple(benchmark_cmd), stdout=subprocess.PIPE)
         for line in iter(process.stdout.readline, b''):
-            print(line.decode("utf-8"), end='')
+            print(line.decode("utf-8"), end='', flush=True)
         process.wait()
         if process.returncode != 0:
             print()

benchmark/serializer.py

@@ -65,8 +65,7 @@ def serialize(dataset_name, dataset_path, serialized_graph_path, benchmark_copy_
             filename = table_types[copy_match.group(1)] + '-' + copy_match.group(1).replace('_', '-') + '_log.txt'
             with open(os.path.join(benchmark_copy_log_dir, filename), 'ab') as f:
                 for line in iter(process.stdout.readline, b''):
-                    sys.stdout.write(line.decode("utf-8"))
-                    sys.stdout.flush()
+                    print(line.decode("utf-8"), end='', flush=True)
                     f.write(line)
         process.wait()
         if process.returncode != 0: