fix

MIC-DKFZ · Dec 28, 2023 · 4d67790 · 4d67790
1 parent 15e33bc
commit 4d67790
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nnunetv2/model_sharing/entry_points.py b/nnunetv2/model_sharing/entry_points.py
@@ -61,7 +61,7 @@ def export_pretrained_model_entry():
                             export_crossval_predictions=args.exp_cv_preds)
 
 
-def export_model_checkpoint():
+def export_model_to_checkpoint():
     import argparse
     parser = argparse.ArgumentParser(description="Export nnunet model checkpoint as a single .pth file")
     parser.add_argument("--path", type=str, help="path to nnunet model directory")

diff --git a/pyproject.toml b/pyproject.toml
@@ -74,7 +74,7 @@ nnUNetv2_plot_overlay_pngs = "nnunetv2.utilities.overlay_plots:entry_point_gener
 nnUNetv2_download_pretrained_model_by_url = "nnunetv2.model_sharing.entry_points:download_by_url"
 nnUNetv2_install_pretrained_model_from_zip = "nnunetv2.model_sharing.entry_points:install_from_zip_entry_point"
 nnUNetv2_export_model_to_zip = "nnunetv2.model_sharing.entry_points:export_pretrained_model_entry"
-nnUNetv2_export_model_to_checkpoint = "nnunetv2.model_sharing.entry_points:export_model_checkpoint"
+nnUNetv2_export_model_to_checkpoint = "nnunetv2.model_sharing.entry_points:export_model_to_checkpoint"
 nnUNetv2_move_plans_between_datasets = "nnunetv2.experiment_planning.plans_for_pretraining.move_plans_between_datasets:entry_point_move_plans_between_datasets"
 nnUNetv2_evaluate_folder = "nnunetv2.evaluation.evaluate_predictions:evaluate_folder_entry_point"
 nnUNetv2_evaluate_simple = "nnunetv2.evaluation.evaluate_predictions:evaluate_simple_entry_point"