From 29b3f07e46f3a0cf527825317b94397de1df8f03 Mon Sep 17 00:00:00 2001 From: Wenqi Li Date: Fri, 8 Oct 2021 15:57:51 +0100 Subject: [PATCH] oneof doc Signed-off-by: Wenqi Li --- docs/source/transforms.rst | 5 +++++ monai/transforms/compose.py | 2 +- 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/docs/source/transforms.rst b/docs/source/transforms.rst index b0c25ddfdb..65c1f7efb6 100644 --- a/docs/source/transforms.rst +++ b/docs/source/transforms.rst @@ -53,6 +53,11 @@ Generic Interfaces .. autoclass:: Decollated :members: +`OneOf` +^^^^^^^ +.. autoclass:: OneOf + :members: + Vanilla Transforms ------------------ diff --git a/monai/transforms/compose.py b/monai/transforms/compose.py index 38aa2199c6..2405530ef3 100644 --- a/monai/transforms/compose.py +++ b/monai/transforms/compose.py @@ -181,7 +181,7 @@ class OneOf(Compose): weights: probabilities corresponding to each callable in transforms. Probabilities are normalized to sum to one. - OneOf inherits from Compose and uses args map_items and unpack_items in + ``OneOf`` inherits from ``Compose`` and uses args ``map_items`` and ``unpack_items`` in the same way. """