diff --git a/training/docs/modules/schemas.rst b/training/docs/modules/schemas.rst
index 179ac061..8aa76ab0 100644
--- a/training/docs/modules/schemas.rst
+++ b/training/docs/modules/schemas.rst
@@ -32,6 +32,15 @@ files,
    :no-undoc-members:
    :show-inheritance:
 
+*************
+ Diagnostics
+*************
+
+.. automodule:: anemoi.training.schemas.diagnostics
+   :members:
+   :no-undoc-members:
+   :show-inheritance:
+
 **********
  Hardware
 **********