diff --git a/src/otx/cli/tools/train.py b/src/otx/cli/tools/train.py
index 8cf27066268..5cf62ec2118 100644
--- a/src/otx/cli/tools/train.py
+++ b/src/otx/cli/tools/train.py
@@ -19,7 +19,6 @@
 import datetime
 import time
 from contextlib import ExitStack
-from functools import partial
 from pathlib import Path
 from typing import Optional