Error in cleaning after rigid registration: 'Valis' object has no attribute 'non_rigid_reg_kwargs'

[maxime915]

Hello,

I'm facing an issue when performing rigid registration (it doesn't happen in non-rigid/micro registration), which happens during the .cleanup method.

Measuring error: 100%|██████████| 2/2 [00:00<00:00, 1090.70image/s]
Traceback (most recent call last):
  File "/root/.local/lib/python3.12/site-packages/valis/registration.py", line 4763, in register
    self.cleanup()
  File "/root/.local/lib/python3.12/site-packages/valis/registration.py", line 4791, in cleanup
    self.non_rigid_reg_kwargs[NON_RIGID_REG_CLASS_KEY] = None
    ^^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'Valis' object has no attribute 'non_rigid_reg_kwargs'. Did you mean: '_set_rigid_reg_kwargs'?

/root/.local/lib/python3.12/site-packages/valis/valtils.py:29: UserWarning: 'Valis' object has no attribute 'non_rigid_reg_kwargs'
  warnings.warn(warning_msg, warning_type)
JVM has been killed. If this was due to an error, then a new Python session will need to be started

This is the line which causes an issue.

valis/valis/registration.py

Line 4791 in 325828c

self.non_rigid_reg_kwargs[NON_RIGID_REG_CLASS_KEY] = None

Because .cleanup is called in .register but non_rigid_reg_kwargs is only set if the registration is non-rigid.

valis/valis/registration.py

Line 4763 in 325828c

self.cleanup()

I'd be happy to submit a PR that checks if those attribute exists before cleaning the registrar, if you're OK with that.

[cdgatenbee]

Hi @maxime915,
Thanks for this, and so sorry it took me this long to respond, I seem to have missed this issue. It would be great if you could
submit a PR to fix this. Thanks again!

Best,
-Chandler