diff --git a/test-requirements.txt b/test-requirements.txt
index 9cf1689..34bbaab 100644
--- a/test-requirements.txt
+++ b/test-requirements.txt
@@ -3,6 +3,7 @@ pytest-cov
 sphinx
 sphinxcontrib-bibtex
 sphinx-paramlinks
+sphinx-autodoc-typehints
 pydata-sphinx-theme>=0.8.0
 matplotlib
 scipy>=1.6.0
@@ -15,4 +16,4 @@ torch
 pykeops
 hnswlib
 eagerpy
-networkx
\ No newline at end of file
+networkx