diff --git a/src/dataset.py b/src/dataset.py
index b0d8aff..0c65904 100644
--- a/src/dataset.py
+++ b/src/dataset.py
@@ -15,7 +15,6 @@ from tenacity import (
     wait_random_exponential,
 )  # for exponential backoff
 import tiktoken
-from time import sleep
 
 from pathlib import Path
 import config