fix: doc

Signed-off-by: ThibaultFy <[email protected]>
Substra · Feb 27, 2024 · 104e738 · 104e738
1 parent 60a9156
commit 104e738
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/docs/api/nodes.rst b/docs/api/nodes.rst
@@ -14,13 +14,17 @@ TestDataNode
 ^^^^^^^^^^^^^
 .. autoclass:: substrafl.nodes.test_data_node.TestDataNode
 
-Node
-^^^^^
-.. autoclass:: substrafl.nodes.node.Node
+Protocols
+^^^^^^^^^
+.. autoclass:: substrafl.nodes.protocol.TrainDataNodeProtocol
+.. autoclass:: substrafl.nodes.protocol.TestDataNodeProtocol
+.. autoclass:: substrafl.nodes.protocol.AggregationNodeProtocol
 
 References
 ^^^^^^^^^^
 .. automodule:: substrafl.nodes.references
-.. autoclass:: substrafl.nodes.node.OperationKey
+.. autoclass:: substrafl.nodes.schemas.OperationKey
+.. autoclass:: substrafl.nodes.schemas.InputIdentifiers
+.. autoclass:: substrafl.nodes.schemas.OutputIdentifiers
 .. autoclass:: substrafl.nodes.references.local_state.LocalStateRef
 .. autoclass:: substrafl.nodes.references.shared_state.SharedStateRef