AttributeError: module 'augment' has no attribute 'effects'

[LYPinASR]

augment/effects.py::augment.effects.EffectChain PASSED [ 7%]
augment/effects.py::augment.effects.EffectChain.additive_noise PASSED [ 14%]
augment/effects.py::augment.effects.EffectChain.clip PASSED [ 21%]
augment/effects.py::augment.effects.EffectChain.time_dropout PASSED [ 28%]
augment/effects.py::augment.effects.SoxEffect.instantiate PASSED [ 35%]
tests/augment_test.py::test_empty_chain PASSED [ 42%]
tests/augment_test.py::test_non_empty_chain PASSED [ 50%]
tests/augment_test.py::test_agains_cl PASSED [ 57%]
tests/augment_test.py::test_stochastic_pitch PASSED [ 64%]
tests/augment_test.py::test_additive_noise PASSED [ 71%]
tests/augment_test.py::test_number_effects PASSED [ 78%]
tests/compare_to_sox_test.py::test_pitch PASSED [ 85%]
tests/compare_to_sox_test.py::test_reverb PASSED [ 92%]
tests/compare_to_sox_test.py::test_bandreject PASSED [100%]

=================================================== 14 passed in 1.71s ====================================================
(torch) root@autodl-container-a1d611953c-bbb3ae7a:/autodl-tmp/WavAugment# cd ../
(torch) root@autodl-container-a1d611953c-bbb3ae7a:/autodl-tmp# python disentanglement/DSVAE/preprocess.py
data processing!
Traceback (most recent call last):
File "disentanglement/DSVAE/preprocess.py", line 27, in
speech_pitch = augment.EffectChain().pitch(-200).rate(sr).apply(speech, src_info={'rate': sr})
AttributeError: module 'augment' has no attribute 'EffectChain'