diff --git a/litgpt/pretrain.py b/litgpt/pretrain.py
index b975450f49..ad37f1262c 100644
--- a/litgpt/pretrain.py
+++ b/litgpt/pretrain.py
@@ -4,7 +4,6 @@
 import os
 import time
 import pprint
-from datetime import timedelta
 from functools import partial
 from pathlib import Path
 from typing import Optional, Tuple, Union