update docs

xarray-contrib · Feb 23, 2022 · 8bcd870 · 8bcd870
1 parent 69909b4
commit 8bcd870
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/doc/api.rst b/doc/api.rst
@@ -5,12 +5,6 @@ API reference
 
 This page provides an auto-generated summary of Xbatcher's API.
 
-Core
-====
-
-.. autoclass:: xbatcher.BatchGenerator
-   :members:
-
 Dataset.batch and DataArray.batch
 =================================
 
@@ -22,3 +16,17 @@ Dataset.batch and DataArray.batch
 
    Dataset.batch.generator
    DataArray.batch.generator
+
+Core
+====
+
+.. autoclass:: xbatcher.BatchGenerator
+   :members:
+
+Dataloaders
+===========
+.. autoclass:: xbatcher.loaders.torch.MapDataset
+   :members:
+
+.. autoclass:: xbatcher.loaders.torch.IterableDataset
+   :members: