Skip to content
Snippets Groups Projects
Commit cf28368b authored by Daniel CARRON's avatar Daniel CARRON :b:
Browse files

Update datamodule docstrings

parent 367ca62f
No related branches found
No related tags found
No related merge requests found
Pipeline #76503 failed
...@@ -95,17 +95,17 @@ def list(): ...@@ -95,17 +95,17 @@ def list():
@datamodule.command( @datamodule.command(
epilog="""Examples: epilog="""Examples:
1. Check if all files of the Montgomery dataset can be loaded: 1. Check if all files from the fold_0 of the Montgomery database can be loaded:
.. code:: sh .. code:: sh
ptbench dataset check -vv shenzhen ptbench datamodule check -vv montgomery_f0
2. Check if all files of multiple installed datasets can be loaded: 2. Check if all files of multiple installed protocols can be loaded:
.. code:: sh .. code:: sh
ptbench dataset check -vv montgomery shenzhen ptbench datamodule check -vv montgomery shenzhen
""", """,
) )
...@@ -116,7 +116,7 @@ def list(): ...@@ -116,7 +116,7 @@ def list():
@click.option( @click.option(
"--limit", "--limit",
"-l", "-l",
help="Limit check to the first N samples in each dataset, making the " help="Limit check to the first N samples in each datamodule, making the "
"check sensibly faster. Set it to zero to check everything.", "check sensibly faster. Set it to zero to check everything.",
required=True, required=True,
type=click.IntRange(0), type=click.IntRange(0),
...@@ -129,6 +129,7 @@ def check(protocols, limit): ...@@ -129,6 +129,7 @@ def check(protocols, limit):
errors = 0 errors = 0
for protocol in protocols: for protocol in protocols:
logger.info(f"Checking {protocol}")
try: try:
module = importlib.metadata.entry_points(group="ptbench.config")[ module = importlib.metadata.entry_points(group="ptbench.config")[
protocol protocol
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment