fix download bug

microsoft · Jul 12, 2024 · 9854d59 · 9854d59
1 parent c47cd6b
commit 9854d59
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/batteryml/preprocess/download.py b/batteryml/preprocess/download.py
@@ -95,12 +95,16 @@ def memory2str(mem):
                     counter += 1
                     bar.update(len(data))
                     if counter % update_interval == 0:
-                        ellapsed = time.time() - now_time
+                        elapsed = time.time() - now_time
                         runtime_downloaded = downloaded - now_size
                         now_time, now_size = time.time(), downloaded
 
                         cur_size = memory2str(downloaded)
-                        speed_size = memory2str(runtime_downloaded / ellapsed)
+                        if elapsed > 0:
+                            speed_size = memory2str(
+                                runtime_downloaded / elapsed)
+                        else:
+                            speed_size = '0B/s'
                         bar.set_description(f'{cur_size}/{total_size}')
                         bar.set_postfix_str(f'{speed_size}/s')