From bff023c1697488be6a28d5f2bd6ea31986fbc8eb Mon Sep 17 00:00:00 2001 From: Qin Yu Date: Wed, 18 Dec 2024 03:05:55 +0100 Subject: [PATCH] fix: prediction widget is not updated by import --- plantseg/viewer_napari/widgets/io.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/plantseg/viewer_napari/widgets/io.py b/plantseg/viewer_napari/widgets/io.py index 7a3d689d..f9bedfa9 100644 --- a/plantseg/viewer_napari/widgets/io.py +++ b/plantseg/viewer_napari/widgets/io.py @@ -14,6 +14,7 @@ from plantseg.tasks.io_tasks import export_image_task, import_image_task from plantseg.tasks.workflow_handler import workflow_handler from plantseg.viewer_napari import log +from plantseg.viewer_napari.widgets.prediction import widget_unet_prediction from plantseg.viewer_napari.widgets.utils import _return_value_if_widget, schedule_task current_dataset_keys: list[str] | None = None @@ -101,7 +102,10 @@ def widget_open_file( elif layer_type == ImageType.LABEL.value: semantic_type = SemanticType.SEGMENTATION - widgets_to_update = [widget_set_voxel_size.layer] + widgets_to_update = [ + widget_set_voxel_size.layer, + widget_unet_prediction.image, + ] return schedule_task( import_image_task,