adding in retry decorator for Batch notebook