module Spruz::Attempt

def sleep_duration(duration, count)
  case duration
  when Numeric
    sleep duration
  when Proc
    sleep duration.call(count)
  end
end