diff --git a/scripts/lpmm_learning_tool.py b/scripts/lpmm_learning_tool.py
index 884b4e9e3..19b2baf7c 100644
--- a/scripts/lpmm_learning_tool.py
+++ b/scripts/lpmm_learning_tool.py
@@ -5,6 +5,7 @@ import shutil
 import sys
 from pathlib import Path
 from threading import Lock
+from concurrent.futures import ThreadPoolExecutor, as_completed
 
 import orjson
 from json_repair import repair_json