Merge pull request #439 from mezotaken/oneof_fix

Fix OneOf so it could choose random augmentation on each subsequent call, instead of choosing once on initialization.

augraphy/base/oneof.py

@@ -18,14 +18,14 @@ class OneOf(Augmentation):
     def __init__(self, augmentations, p=1):
         """Constructor method"""
         self.augmentations = augmentations
-        self.augmentation_probabilities = self.compute_probability(self.augmentations)
         self.p = p
 
     # Randomly selects an Augmentation to apply to data.
     def __call__(self, image, layer=None, mask=None, keypoints=None, bounding_boxes=None, force=False):
         if force or self.should_run():
 
             # Select one augmentation using the max value in probability values
+            self.augmentation_probabilities = self.compute_probability(self.augmentations)
             augmentation = self.augmentations[np.argmax(self.augmentation_probabilities)]
 
             # Applies the selected Augmentation.