lib/retriable/retriable.rb in retriable-1.3.3 vs lib/retriable/retriable.rb in retriable-1.3.3.1

@@ -33,10 +33,13 @@
       rescue *[*on] => exception
         @tries -= 1
         if @tries > 0
           count += 1
           @on_retry.call(exception, count) if @on_retry
-          sleep @interval if @interval > 0
+
+          sleep_for = @interval.respond_to?(:call) ? @interval.call(count) : @interval
+          sleep sleep_for if sleep_for > 0
+
           retry
         else
           raise
         end
       end