Replacing sys.stdout.write with print in progress bar

ghislainv · Jun 8, 2024 · 9d3b42b · 9d3b42b
1 parent f12cafb
commit 9d3b42b
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 157 deletions.
diff --git a/riskmapjnr/benchmark/defrate_per_class_model.py b/riskmapjnr/benchmark/defrate_per_class_model.py
diff --git a/riskmapjnr/benchmark/vulnerability_map.py b/riskmapjnr/benchmark/vulnerability_map.py
@@ -129,10 +129,11 @@ def vulnerability_map(
     cat_band.ComputeStatistics(False, cb)
 
     # Dereference drivers
+    forest_band = None
     cat_band = None
     dist_band = None
     subj_band = None
-    del cat_ds, dist_ds, subj_ds
+    del forest_ds, cat_ds, dist_ds, subj_ds
 
 
 # # Test

diff --git a/riskmapjnr/misc/miscellaneous.py b/riskmapjnr/misc/miscellaneous.py
@@ -172,14 +172,17 @@ def progress_bar(niter, i):
 
     step = 1 if niter <= 100 else niter // 100
     if i == 1:
-        sys.stdout.write("0%")
-        sys.stdout.flush()
+        print("0%", end="", flush=True)
+        #sys.stdout.write("0%")
+        #sys.stdout.flush()
     elif i % step == 0:
-        sys.stdout.write("\r{}%".format((100 * i) // niter))
-        sys.stdout.flush()
+        print(f"\r{(100 * i) // niter}%", end="", flush=True)
+        #sys.stdout.write("\r{}%".format((100 * i) // niter))
+        #sys.stdout.flush()
     if i == niter:
-        sys.stdout.write("\r100%\n")
-        sys.stdout.flush()
+        print("\r100%", flush=True)
+        #sys.stdout.write("\r100%\n")
+        #sys.stdout.flush()
 
 
 # Rescale