From 0ec62a0df5dcb2e0f3f261252ee130bb4969a35c Mon Sep 17 00:00:00 2001 From: luandalmazo Date: Mon, 6 Jul 2026 14:22:19 -0300 Subject: [PATCH] update nnunet to log annotations specs --- datamint/lightning/trainers/specialized/nnunet/trainer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/datamint/lightning/trainers/specialized/nnunet/trainer.py b/datamint/lightning/trainers/specialized/nnunet/trainer.py index 65fc90b7..c2b62ccf 100644 --- a/datamint/lightning/trainers/specialized/nnunet/trainer.py +++ b/datamint/lightning/trainers/specialized/nnunet/trainer.py @@ -549,4 +549,5 @@ def _build_deploy_adapter(self, dataset_id: int, bridge) -> None: name='nnunet_model', artifacts={'nnunet_bundle': str(bundle)}, extra_pip_requirements=[f'nnunetv2=={_nnunet_ver}'], + annotation_specs=self._build_annotation_specs(), )