Skip to content

Commit 3b5e80b

Browse files
Update documentation for nnUNet Bundle integration
1 parent c107015 commit 3b5e80b

File tree

2 files changed

+12
-10
lines changed

2 files changed

+12
-10
lines changed

docs/source/apps.rst

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -277,3 +277,14 @@ FastMRIReader
277277

278278
.. autoclass:: monai.apps.nnunet.nnUNetV2Runner
279279
:members:
280+
281+
`nnUNet Bundle`
282+
---------------
283+
.. autoclass:: monai.apps.nnunet.ModelnnUNetWrapper
284+
:members:
285+
:special-members:
286+
287+
.. autofunction:: monai.apps.nnunet.get_nnunet_trainer
288+
.. autofunction:: monai.apps.nnunet.get_nnunet_monai_predictor
289+
.. autofunction:: monai.apps.nnunet.convert_nnunet_to_monai_bundle
290+

docs/source/bundle.rst

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -35,12 +35,6 @@ Model Bundle
3535
:special-members:
3636

3737

38-
`nnUNet Bundle`
39-
---------------
40-
.. autoclass:: ModelnnUNetWrapper
41-
:members:
42-
:special-members:
43-
4438
`Scripts`
4539
---------
4640
.. autofunction:: ckpt_export
@@ -56,7 +50,4 @@ Model Bundle
5650
.. autofunction:: verify_net_in_out
5751
.. autofunction:: init_bundle
5852
.. autofunction:: push_to_hf_hub
59-
.. autofunction:: update_kwargs
60-
.. autofunction:: get_nnunet_trainer
61-
.. autofunction:: get_nnunet_monai_predictor
62-
.. autofunction:: convert_nnunet_to_monai_bundle
53+
.. autofunction:: update_kwargs

0 commit comments

Comments
 (0)