diff --git a/src/ptbench/scripts/click.py b/src/ptbench/scripts/click.py index 39cf96f962fced761bb68c12a593a24be19e9fa5..07dfe697c5f0e35356f75958c949c97aaef36aa8 100644 --- a/src/ptbench/scripts/click.py +++ b/src/ptbench/scripts/click.py @@ -27,4 +27,4 @@ class ConfigCommand(_BaseConfigCommand): if self.epilog: formatter.write_paragraph() for line in self.epilog.split("\n"): - formatter.write_text(line) + formatter.write(line + "\n") diff --git a/src/ptbench/scripts/generate_saliencymaps.py b/src/ptbench/scripts/generate_saliencymaps.py index a7232fedc2bebca15b6809eda422ec17b0ac55e4..7ba30ede63036a36b17c230677d9aabd125a48d4 100644 --- a/src/ptbench/scripts/generate_saliencymaps.py +++ b/src/ptbench/scripts/generate_saliencymaps.py @@ -7,10 +7,11 @@ import typing import click -from clapper.click import ConfigCommand, ResourceOption, verbosity_option +from clapper.click import ResourceOption, verbosity_option from clapper.logging import setup from ..models.typing import VisualisationType +from .click import ConfigCommand logger = setup(__name__.split(".")[0], format="%(levelname)s: %(message)s") @@ -20,15 +21,13 @@ logger = setup(__name__.split(".")[0], format="%(levelname)s: %(message)s") cls=ConfigCommand, epilog="""Examples: -\b - 1. Generates saliency maps for all prediction dataloaders on a datamodule, - using a pre-trained DenseNet model, and saves them as numpy-pickeled - objects on the output directory: +1. Generates saliency maps for all prediction dataloaders on a datamodule, + using a pre-trained DenseNet model, and saves them as numpy-pickeled + objects on the output directory: - .. code:: sh + .. code:: sh -\b - ptbench generate-saliencymaps -vv densenet tbx11k-v1-healthy-vs-atb --weight=path/to/model_final.pth --output-folder=path/to/visualisations + ptbench generate-saliencymaps -vv densenet tbx11k-v1-healthy-vs-atb --weight=path/to/model_final.pth --output-folder=path/to/visualisations """, )