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/_code.rst
+++ b/docs/source/_rst/_code.rst
@@ -26,6 +26,7 @@ Trainer, Dataset and Datamodule
    Trainer <trainer.rst>
    Dataset <data/dataset.rst>
    DataModule <data/data_module.rst>
+    Dataloader <data/dataloader.rst>

 Data Types
 ------------
--- 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: