Dynamically generate RVTransform list in docs

Author: rlouf

docs/source/api/transforms.rst

@@ -56,37 +56,14 @@ associated variables and their values:
 
 .. autoclass:: aeppl.transforms.TransformValuesRewrite
 
+
 ---------------------
 Invertible transforms
 ---------------------
 
 AePPL currently supports transforms using the following (invertible) Aesara operators. This means that AePPL can compute the log-probability of a random variable that is the result of one of the following transformations applied to another random variable:
 
-- `aesara.tensor.add`
-- `aesara.tensor.sub`
-- `aesara.tensor.mul`
-- `aesara.tensor.true_div`
-- `aesara.tensor.exponential`
-- `aesara.tensor.exp`
-- `aesara.tensor.log`
-
-One can also apply the following transforms directly:
-
-.. autoclass:: aeppl.transforms.LocTransform
-.. autoclass:: aeppl.transforms.ScaleTransform
-.. autoclass:: aeppl.transforms.LogTransform
-.. autoclass:: aeppl.transforms.ExpTransform
-.. autoclass:: aeppl.transforms.ReciprocalTransform
-.. autoclass:: aeppl.transforms.IntervalTransform
-.. autoclass:: aeppl.transforms.LogOddsTransform
-.. autoclass:: aeppl.transforms.SimplexTransform
-.. autoclass:: aeppl.transforms.CircularTransform
-
-These transformations can be chained using:
-
-
-.. autoclass:: aeppl.transforms.ChainedTransform
-
+.. print-invertible-transforms::
 
 ---------
 Censoring

docs/source/conf.py

@@ -112,5 +112,42 @@ def run(self):
         return [res]
 
 
+class InvertibleTransformationsDirective(SphinxDirective):
+    def run(self):
+        import inspect
+        import sys
+
+        import aeppl.transforms as transforms
+
+        invertible_transforms = (
+            mname
+            for mname, mtype in inspect.getmembers(sys.modules["aeppl.transforms"])
+            if inspect.isclass(mtype)
+            and issubclass(mtype, transforms.RVTransform)
+            and not mtype == transforms.RVTransform
+        )
+
+        res = nodes.bullet_list()
+        for transform_name in invertible_transforms:
+            attributes = {}
+            reftarget = f"aeppl.transforms.{transform_name}"
+            attributes["reftarget"] = reftarget
+            attributes["reftype"] = "class"
+            attributes["refdomain"] = "py"
+
+            ref = nodes.paragraph()
+
+            rawsource = rf":py:class:`{reftarget}`"
+            xref = sphinx.addnodes.pending_xref(rawsource, **attributes)
+            xref += nodes.literal(reftarget, reftarget, classes=["xref"])
+
+            ref += xref
+            item = nodes.list_item("", ref)
+            res += item
+
+        return [res]
+
+
 def setup(app):
     app.add_directive("print-supported-dists", SupportedDistributionsDirective)
+    app.add_directive("print-invertible-transforms", InvertibleTransformationsDirective)