[Datasets] Add missing public APIs to Datasets API docs (ray-project#…

…23935)
jbn · Apr 16, 2022 · 57f620b · 57f620b
1 parent b06bab8
commit 57f620b
Showing 1 changed file with 39 additions and 0 deletions.
diff --git a/doc/source/data/package-ref.rst b/doc/source/data/package-ref.rst
@@ -38,6 +38,24 @@ Dataset API
 
 .. _dataset-pipeline-api:
 
+Block API
+---------
+
+.. autoclass:: ray.data.block.BlockExecStats
+    :members:
+
+.. autoclass:: ray.data.block.BlockMetadata
+    :members:
+
+.. autoclass:: ray.data.block.BlockAccessor
+    :members:
+
+DatasetContext API
+------------------
+
+.. autoclass:: ray.data.context.DatasetContext
+    :members:
+
 DatasetPipeline API
 -------------------
 
@@ -50,6 +68,27 @@ GroupedDataset API
 .. autoclass:: ray.data.grouped_dataset.GroupedDataset
     :members:
 
+Aggregate API
+-------------
+
+.. autoclass:: ray.data.aggregate.AggregateFn
+    :members:
+
+.. autoclass:: ray.data.aggregate.Count
+    :members:
+
+.. autoclass:: ray.data.aggregate.Sum
+    :members:
+
+.. autoclass:: ray.data.aggregate.Max
+    :members:
+
+.. autoclass:: ray.data.aggregate.Mean
+    :members:
+
+.. autoclass:: ray.data.aggregate.Std
+    :members:
+
 RandomAccessDataset API
 -----------------------