Merge branch 'master' of ssh://phabricator.mitk.org:2222/source/dldabg

Karol-G · Feb 23, 2023 · 61b4334 · 61b4334
2 parents e59c73c + 7651ece
commit 61b4334
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/batchgenerators/augmentations/spatial_transformations.py b/batchgenerators/augmentations/spatial_transformations.py
@@ -298,7 +298,7 @@ def augment_spatial_2(data, seg, patch_size, patch_center_dist_from_border=30,
                       do_scale=True, scale=(0.75, 1.25), border_mode_data='nearest', border_cval_data=0, order_data=3,
                       border_mode_seg='constant', border_cval_seg=0, order_seg=0, random_crop=True, p_el_per_sample=1,
                       p_scale_per_sample=1, p_rot_per_sample=1, independent_scale_for_each_axis=False,
-                      p_rot_per_axis: float = 1, p_independent_scale_per_axis: int = 1):
+                      p_rot_per_axis: float = 1, p_independent_scale_per_axis: float = 1):
     """
 
     :param data:

diff --git a/batchgenerators/transforms/spatial_transforms.py b/batchgenerators/transforms/spatial_transforms.py
@@ -417,7 +417,7 @@ def __init__(self, patch_size, patch_center_dist_from_border=30,
                  do_scale=True, scale=(0.75, 1.25), border_mode_data='nearest', border_cval_data=0, order_data=3,
                  border_mode_seg='constant', border_cval_seg=0, order_seg=0, random_crop=True, data_key="data",
                  label_key="seg", p_el_per_sample=1, p_scale_per_sample=1, p_rot_per_sample=1,
-                 independent_scale_for_each_axis=False, p_rot_per_axis:float=1, p_independent_scale_per_axis: int=1):
+                 independent_scale_for_each_axis=False, p_rot_per_axis:float=1, p_independent_scale_per_axis: float=1):
         self.p_rot_per_sample = p_rot_per_sample
         self.p_scale_per_sample = p_scale_per_sample
         self.p_el_per_sample = p_el_per_sample

diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 setup(name='batchgenerators',
-      version='0.23',
+      version='0.24',
       description='Data augmentation toolkit',
       url='https://github.com/MIC-DKFZ/batchgenerators',
       author='Division of Medical Image Computing, German Cancer Research Center AND Applied Computer Vision Lab, '