diff --git a/docs/conf.py b/docs/conf.py
index c432122..7839614 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -35,6 +35,8 @@
     'sphinx.ext.napoleon',
     'sphinx_autodoc_typehints',
     'sphinx.ext.viewcode',
+    'sphinx.ext.inheritance_diagram',
+    'sphinx.ext.graphviz'
 ]
 
 # Add any paths that contain templates here, relative to this directory.