Fix for rios v2.0 in how the progress bar is called.

remotesensinginfo · Jun 13, 2024 · e7b2296 · e7b2296
1 parent a7d0eb7
commit e7b2296
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/rsgislib/__init__.py b/python/rsgislib/__init__.py
@@ -698,6 +698,7 @@ class TQDMProgressBar:
 
     def __init__(self):
         self.lprogress = 0
+        self.pbar = None
 
     def setTotalSteps(self, steps: int):
         import tqdm
@@ -711,13 +712,16 @@ def setTotalSteps(self, steps: int):
         self.lprogress = 0
 
     def setProgress(self, progress: int):
+        if self.pbar is None:
+            self.setTotalSteps(steps=100)
         step = progress - self.lprogress
         self.pbar.update(step)
         self.lprogress = progress
 
     def reset(self):
-        self.pbar.close()
         import tqdm
+        if self.pbar is not None:
+            self.pbar.close()
 
         if is_notebook():
             import tqdm.notebook