diff --git a/requirements.txt b/requirements.txt
index 30477dc..53c67d7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -20,4 +20,5 @@ pyparsing<3
 nvidia-pyindex
 onnx-graphsurgeon
 ddt
-pymongo
\ No newline at end of file
+pymongo
+sklearn
\ No newline at end of file