retry tqdm openai torch datasets transformers scikit-learn