Minor fixes for Topological Regularizer

[JairMathAI]

I have made some minor fixes to enable correct training with both 2D and 3D images for the Probabilistic U-Net architecture. I also included a more robust implementation of the model to be used with regularizers. I am keeping the previous version for now to maintain compatibility with the SV project so Phillip can still run predictions; however, once the latest models are finalized, this older implementation will be removed.

[codecov-commenter]

Codecov Report

❌ Patch coverage is 2.10526% with 93 lines in your changes missing coverage. Please review.
✅ Project coverage is 17.15%. Comparing base (ba85ae2) to head (1b5bf03).
⚠️ Report is 31 commits behind head on main.

Files with missing lines	Patch %	Lines
mmv_im2im/models/pl_ProbUnet_old.py	0.00%	81 Missing ⚠️
mmv_im2im/map_extractor.py	0.00%	4 Missing ⚠️
mmv_im2im/models/pl_ProbUnet.py	0.00%	4 Missing ⚠️
mmv_im2im/models/nets/ProbUnet.py	0.00%	2 Missing ⚠️
mmv_im2im/postprocessing/basic_collection.py	0.00%	2 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##            main     #116       +/-   ##
==========================================
+ Coverage   4.40%   17.15%   +12.75%     
==========================================
  Files         36       51       +15     
  Lines       2815     5252     +2437     
  Branches       0      847      +847     
==========================================
+ Hits         124      901      +777     
- Misses      2691     4208     +1517     
- Partials       0      143      +143     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.