diff --git a/scripts/train/train.py b/scripts/train/train.py
index 5a05d18e94..4620254135 100644
--- a/scripts/train/train.py
+++ b/scripts/train/train.py
@@ -50,7 +50,6 @@
     make_dataclass_and_log_config,
     pop_config,
     process_init_device,
-    update_batch_size_info,
 )
 from llmfoundry.utils.exceptions import (
     BaseContextualError,