diff --git a/python/rapid_structure/requirements.txt b/python/rapid_structure/requirements.txt
index d847c7d..279256a 100644
--- a/python/rapid_structure/requirements.txt
+++ b/python/rapid_structure/requirements.txt
@@ -4,4 +4,5 @@ setuptools>=61.2.0
 PyYAML
 pytest>=7.2.1
 pip==22.3.1
-onnxruntime
\ No newline at end of file
+onnxruntime
+wheel
\ No newline at end of file