From d75339c69962c146c4a0f76c7b71d6c06d6f61a6 Mon Sep 17 00:00:00 2001 From: Maxime Amodei Date: Fri, 12 Dec 2025 15:55:57 +0100 Subject: [PATCH] fix: only cleanup non_rigid_reg_kwargs if it is defined --- valis/registration.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/valis/registration.py b/valis/registration.py index fb98c30b..6c0560bd 100644 --- a/valis/registration.py +++ b/valis/registration.py @@ -4788,7 +4788,8 @@ def cleanup(self): self.rigid_reg_kwargs[AFFINE_OPTIMIZER_KEY] = None self.rigid_reg_kwargs[MATCHER_KEY] = None self.rigid_reg_kwargs[MATCHER_FOR_SORTING_KEY] = None - self.non_rigid_reg_kwargs[NON_RIGID_REG_CLASS_KEY] = None + if hasattr(self, "non_rigid_reg_kwargs"): + self.non_rigid_reg_kwargs[NON_RIGID_REG_CLASS_KEY] = None self.non_rigid_registrar_cls = None self.rigid_registrar = None self.micro_rigid_registrar_cls = None