Cleanup graph API docs

ricardoV94 · ricardoV94 · commit 8c3e68f97c70 · 2024-07-05T12:11:00.000+02:00
diff --git a/doc/library/graph/graph.rst b/doc/library/graph/graph.rst
@@ -4,12 +4,5 @@
 :mod:`graph` -- Interface for the PyTensor graph
 ================================================
 
----------
-Reference
----------
-
 .. automodule:: pytensor.graph.basic
-   :platform: Unix, Windows
-   :synopsis: Interface for types of symbolic variables
    :members:
-.. moduleauthor:: LISA
diff --git a/doc/library/graph/index.rst b/doc/library/graph/index.rst
@@ -1,20 +1,20 @@
 
 .. _libdoc_graph:
 
-================================================
-:mod:`graph` -- Theano Internals [doc TODO]
-================================================
+========================================
+:mod:`graph` -- PyTensor Graph Internals
+========================================
 
 .. module:: graph
-   :platform: Unix, Windows
-   :synopsis: Theano Internals
+
 .. moduleauthor:: LISA
 
 .. toctree::
     :maxdepth: 1
 
     graph
     fgraph
+    replace
     features
     op
     type
diff --git a/doc/library/graph/op.rst b/doc/library/graph/op.rst
@@ -1,12 +1,8 @@
-
 .. _libdoc_graph_op:
 
-==============================================================
-:mod:`graph` -- Objects and functions for computational graphs
-==============================================================
+===========================================
+:mod:`op` -- Objects that define operations
+===========================================
 
 .. automodule:: pytensor.graph.op
-   :platform: Unix, Windows
-   :synopsis: Interface for types of symbolic variables
-   :members:
-.. moduleauthor:: LISA
+   :members:
diff --git a/doc/library/graph/replace.rst b/doc/library/graph/replace.rst
@@ -0,0 +1,8 @@
+.. _libdoc_graph_replace:
+
+==================================================
+:mod:`replace` -- High level graph transformations
+==================================================
+
+.. automodule:: pytensor.graph.replace
+
