ProgressBar::WithProgress use #size when available

paul · Dec 11, 2024 · 63ed8d7 · 63ed8d7
1 parent 70eb837
commit 63ed8d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/progress_bar/with_progress.rb b/lib/progress_bar/with_progress.rb
@@ -3,7 +3,7 @@
 class ProgressBar
   module WithProgress
     def each_with_progress(*args, &block)
-      bar = ProgressBar.new(count, *args)
+      bar = ProgressBar.new(respond_to?(:size) && size || count, *args)
       if block
         each{ |obj| yield(obj).tap{ bar.increment! } }
       else