ModuleNotFoundError when importing

[katrin-ibrahim]

ModuleNotFoundError Traceback (most recent call last) in () 32 import librosa, librosa.display 33 from audiomentations import Compose, AddGaussianNoise, FrequencyMask, TimeStretch, PitchShift, Shift ---> 34 from specAugment import spec_augment_tensorflow 35 import IPython 36 import soundfile as sf

/usr/local/lib/python3.7/dist-packages/specAugment/spec_augment_tensorflow.py in () 39 import librosa.display 40 import tensorflow as tf ---> 41 from tensorflow.contrib.image import sparse_image_warp 42 import numpy as np 43 import random

ModuleNotFoundError: No module named 'tensorflow.contrib'

[rockamondo]

this code is using an older version of tensorflow, here are some edits i made to spec_augment_tensorflow.py to make it compatible with tensorflow 2.9.1:

change line 40 import statement "import tensorflow as tf" to "import tensorflow.compat.v1 as tf" change line 41 import statement "from tensorflow.contrib.image import sparse_image_warp" to "from tensorflow_addons.image import sparse_image_warp"

This still threw an error about "eager execution" wrt the "placeholder" function when trying to run the "spec_augment" function. I fixed it by inserting the command "tf.disable_eager_execution" on line 75 before the call of the placeholder function.