diff --git a/requirements.docs.txt b/requirements.docs.txt
index a214169..6a5f733 100644
--- a/requirements.docs.txt
+++ b/requirements.docs.txt
@@ -1,3 +1,4 @@
+-e file:.[training]
 mkdocs-material==9.5.3
 mkdocstrings==0.24.0
 mkdocstrings-python==1.8.0