Notebook unit testing

Notion task: https://www.notion.so/qim-dtu/Unit-test-for-notebooks-184bab83b2f980df8b99ebb8c4386c83

• Updated the notebooks to comply with the refactoring changes.
• Implemented unit tests for all notebooks (except annotation_tool.ipynb, segmentation_pipeline.ipynb, and Unet.ipynb). The unit tests are located in qim3d/tests/notebooks/test_notebooks.py.

qim3d/ml/__init__.py

 from ._augmentations import Augmentation
 from ._data import Dataset, prepare_dataloaders, prepare_datasets
-from ._ml_utils import inference, model_summary, train_model
+from ._ml_utils import inference, volume_inference, model_summary, train_model
 from .models import *
\ No newline at end of file