update rst

2025-03-13 16:42:28 +01:00
parent 65534c998a
commit 9a26c94e07
45 changed files with 223 additions and 155 deletions
--- a/docs/source/_rst/data/data_module.rst
+++ b/docs/source/_rst/data/data_module.rst
@@ -1,7 +1,15 @@
-PinaDataModule
+DataModule
 ======================
 .. currentmodule:: pina.data.data_module

+.. autoclass:: Collator
+    :members:
+    :show-inheritance:
+
 .. autoclass:: PinaDataModule
+    :members:
+    :show-inheritance:
+
+.. autoclass:: PinaSampler
    :members:
    :show-inheritance:
--- a/docs/source/_rst/data/dataset.rst
+++ b/docs/source/_rst/data/dataset.rst
@@ -1,7 +1,19 @@
-PinaDataset
+Dataset
 ======================
 .. currentmodule:: pina.data.dataset

 .. autoclass:: PinaDataset
+    :members:
+    :show-inheritance:
+
+.. autoclass:: PinaDatasetFactory
+    :members:
+    :show-inheritance:
+
+.. autoclass:: PinaGraphDataset
+    :members:
+    :show-inheritance:
+
+.. autoclass:: PinaTensorDataset
    :members:
    :show-inheritance: