Update augmentation __init__ with all transforms
Utilities for data augmentation.
"""
from .base import Transform, IdentityTransform, Compose, BaseDataLoader
-from .transforms import TimeReverse, DownsamplingShift, FTSurrogate,\
- ShuffleChannels, MissingChannels
+from .transforms import (
+ TimeReverse, SignFlip, DownsamplingShift, FTSurrogate, ShuffleChannels,
+ MissingChannels, GaussianNoise, ChannelSymmetry, TimeMask,
+ BandstopFilter, FrequencyShift, RandomSensorsRotation, RandomZRotation,
+ RandomYRotation, RandomXRotation,
+)
