diff --git a/requirements.txt b/requirements.txt
index 5ca839a..d9721de 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,4 +13,5 @@ botocore
 huggingface_hub
 mkdocs
 mkdocs-material
-mkdocstrings
\ No newline at end of file
+mkdocstrings
+pytest
\ No newline at end of file