diff --git a/src/mednet/libs/classification/scripts/database.py b/src/mednet/libs/classification/scripts/database.py index b76f2e6987fb22aa8716c701ceae8dae21e5b1ff..038270d7e1c88a791d98f2d4dcf7a490827b27ba 100644 --- a/src/mednet/libs/classification/scripts/database.py +++ b/src/mednet/libs/classification/scripts/database.py @@ -112,7 +112,7 @@ def list_(): """, ) @click.argument( - "split", + "fold", nargs=1, ) @click.option( @@ -125,6 +125,6 @@ def list_(): default=0, ) @verbosity_option(logger=logger, expose_value=False) -def check(split, limit): # numpydoc ignore=PR01 +def check(fold, limit): # numpydoc ignore=PR01 """Check file access on one or more DataModules.""" - check_("mednet.libs.classification.config", split, limit) + check_("mednet.libs.classification.config", fold, limit) diff --git a/src/mednet/libs/common/scripts/database.py b/src/mednet/libs/common/scripts/database.py index 755e6afd351b8069247943d2c1aab6b58739e739..63f1dde1c17c28778288151d0db06fa76ab7a378 100644 --- a/src/mednet/libs/common/scripts/database.py +++ b/src/mednet/libs/common/scripts/database.py @@ -35,18 +35,18 @@ def list_(config: dict[str, dict[str, str]]) -> None: click.echo(f'- {k} ({v["module"]}): NOT installed') -def check(entry_point_group, split, limit): # numpydoc ignore=PR01 +def check(entry_point_group, fold, limit): # numpydoc ignore=PR01 """Check file access on one or more DataModules.""" import importlib.metadata import sys - click.secho(f"Checking split `{split}`...", fg="yellow") + click.secho(f"Checking fold `{fold}`...", fg="yellow") try: module = importlib.metadata.entry_points(group=entry_point_group)[ - split + fold ].module except KeyError: - raise Exception(f"Could not find database split `{split}`") + raise Exception(f"Could not find database fold `{fold}`") datamodule = importlib.import_module(module).datamodule @@ -60,14 +60,14 @@ def check(entry_point_group, split, limit): # numpydoc ignore=PR01 for k, loader in loaders.items(): if limit == 0: click.secho( - f"Checking all samples of dataset `{k}` at split `{split}`...", + f"Checking all samples of dataset `{k}` at fold `{fold}`...", fg="yellow", ) loader_limit = sys.maxsize else: click.secho( f"Checking first {limit} samples of dataset " - f"`{k}` at split `{split}`...", + f"`{k}` at fold `{fold}`...", fg="yellow", ) loader_limit = limit @@ -88,11 +88,11 @@ def check(entry_point_group, split, limit): # numpydoc ignore=PR01 if not errors: click.secho( - f"OK! No errors were reported for database split `{split}`.", + f"OK! No errors were reported for database fold `{fold}`.", fg="green", ) else: click.secho( - f"Found {errors} errors loading DataModule `{split}`.", + f"Found {errors} errors loading DataModule `{fold}`.", fg="red", ) diff --git a/src/mednet/libs/segmentation/scripts/database.py b/src/mednet/libs/segmentation/scripts/database.py index 2d89b4f54cc02a8afa010b87c27f9b7bc9ac5c46..a58bea038150f4fdfa3fd6df569069a4742c2f6c 100644 --- a/src/mednet/libs/segmentation/scripts/database.py +++ b/src/mednet/libs/segmentation/scripts/database.py @@ -101,7 +101,7 @@ def list_(): @database.command( epilog="""Examples: - 1. Check if all files from the split 'montgomery-f0' of the Montgomery + 1. Check if all files from the fold 'montgomery-f0' of the Montgomery database can be loaded: .. code:: sh @@ -111,7 +111,7 @@ def list_(): """, ) @click.argument( - "split", + "fold", nargs=1, ) @click.option( @@ -124,6 +124,6 @@ def list_(): default=0, ) @verbosity_option(logger=logger, expose_value=False) -def check(split, limit): # numpydoc ignore=PR01 +def check(fold, limit): # numpydoc ignore=PR01 """Check file access on one or more DataModules.""" - check_("mednet.libs.segmentation.config", split, limit) + check_("mednet.libs.segmentation.config", fold, limit)