diff --git a/docs/requirements.txt b/docs/requirements.txt
index 2dc53fd..e2d9801 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -19,4 +19,5 @@ https://pytorch-geometric.com/whl/torch-1.7.0/torch_scatter-2.0.5%2Bcpu-cp36-cp3
 https://pytorch-geometric.com/whl/torch-1.7.0/torch_sparse-0.6.8%2Bcpu-cp36-cp36m-linux_x86_64.whl
 https://pytorch-geometric.com/whl/torch-1.7.0/torch_spline_conv-1.2.0%2Bcpu-cp36-cp36m-linux_x86_64.whl
 torch-geometric
-tqdm
\ No newline at end of file
+tqdm
+nni
\ No newline at end of file