fix doc

2025-11-14 17:03:38 +01:00
parent 43163fdf74
commit 0b877d86b9
6 changed files with 20 additions and 20 deletions
--- a/docs/source/_rst/data/data_module.rst
+++ b/docs/source/_rst/data/data_module.rst
@@ -2,14 +2,6 @@ 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/dataloader.rst
+++ b/docs/source/_rst/data/dataloader.rst
@@ -0,0 +1,11 @@
+Dataloader
+======================
+.. currentmodule:: pina.data.dataloader
+
+.. autoclass:: PinaSampler
+    :members:
+    :show-inheritance:
+    
+.. autoclass:: PinaDataLoader
+    :members:
+    :show-inheritance:
--- a/docs/source/_rst/data/dataset.rst
+++ b/docs/source/_rst/data/dataset.rst
@@ -7,13 +7,5 @@ Dataset
    :show-inheritance:

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