Skip to content

Commit

Permalink
Fix PEP8
Browse files Browse the repository at this point in the history
  • Loading branch information
BloodAxe committed Aug 27, 2018
1 parent bb1a089 commit 5d3d786
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions albumentations/imgaug/transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,8 @@
from ..core.transforms_interface import BasicTransform, DualTransform, ImageOnlyTransform

__all__ = ['BasicIAATransform', 'DualIAATransform', 'ImageOnlyIAATransform', 'IAAEmboss', 'IAASuperpixels',
'IAASharpen', 'IAAAdditiveGaussianNoise', 'IAACropAndPad', 'IAAFliplr', 'IAAFlipud', 'IAAAffine', 'IAAPiecewiseAffine', 'IAAPerspective']
'IAASharpen', 'IAAAdditiveGaussianNoise', 'IAACropAndPad', 'IAAFliplr', 'IAAFlipud', 'IAAAffine',
'IAAPiecewiseAffine', 'IAAPerspective']


class BasicIAATransform(BasicTransform):
Expand Down Expand Up @@ -159,7 +160,8 @@ class IAAAffine:
Targets:
image, mask
"""
def __init__(self, scale=1.0, translate_percent=None, translate_px=None, rotate=0.0, shear=0.0, order=1, cval=0, mode="reflect", p=0.5):
def __init__(self, scale=1.0, translate_percent=None, translate_px=None, rotate=0.0, shear=0.0, order=1, cval=0,
mode="reflect", p=0.5):
super(IAAAffine, self).__init__(p)
self.processor = iaa.Affine(scale, translate_percent, translate_px, rotate, shear, order, cval, mode)

Expand Down

0 comments on commit 5d3d786

Please sign in to comment.