From 95a20406ad96d42f798d3b31f26da9466013d436 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Mon, 13 May 2024 12:17:51 +0200
Subject: [PATCH] [segmentation] Add train-analysis command
---
src/mednet/libs/common/scripts/train_analysis.py | 3 ++-
src/mednet/libs/segmentation/scripts/cli.py | 9 ++++++++-
2 files changed, 10 insertions(+), 2 deletions(-)
diff --git a/src/mednet/libs/common/scripts/train_analysis.py b/src/mednet/libs/common/scripts/train_analysis.py
index bf3efe88..65350228 100644
--- a/src/mednet/libs/common/scripts/train_analysis.py
+++ b/src/mednet/libs/common/scripts/train_analysis.py
@@ -6,7 +6,7 @@ import pathlib
import typing
import click
-from clapper.click import ResourceOption, verbosity_option
+from clapper.click import verbosity_option
from clapper.logging import setup
from .click import ConfigCommand
@@ -220,6 +220,7 @@ def _create_figures(
writable=True,
path_type=pathlib.Path,
),
+ default="results",
)
@verbosity_option(logger=logger, expose_value=False)
def train_analysis(
diff --git a/src/mednet/libs/segmentation/scripts/cli.py b/src/mednet/libs/segmentation/scripts/cli.py
index 82aaed41..52ce017f 100644
--- a/src/mednet/libs/segmentation/scripts/cli.py
+++ b/src/mednet/libs/segmentation/scripts/cli.py
@@ -2,6 +2,8 @@
#
# SPDX-License-Identifier: GPL-3.0-or-later
+import importlib
+
import click
from clapper.click import AliasedGroup
@@ -38,4 +40,9 @@ segmentation.add_command(database.database)
# segmentation.add_command(predict.predict)
# segmentation.add_command(significance.significance)
segmentation.add_command(train.train)
-# segmentation.add_command(train_analysis.train_analysis)
+segmentation.add_command(
+ importlib.import_module(
+ "mednet.libs.common.scripts.train_analysis",
+ package=__name__,
+ ).train_analysis,
+)
--
GitLab