Wraps a SoxBindings transformation in such a way that it can be used inside a tf.data map call.

mappable_sox_fn.py

sox_effects = {
    'compand': {},
    'chorus': {},
    'highpass': {'frequency': 100},
    'lowpass': {'frequency': 8000},
    'phaser': {},
    'reverb': {}
}

def get_sox_effect(
    effect_type: str,
    effect_params: dict
    ) -> Callable[[tf.Tensor], np.ndarray]:
    # this allows multi-threading envs
    @sox.sox_context()
    def sox_effect(y: tf.Tensor) -> np.ndarray:
        y = y.numpy()
        tfm = sox.Transformer()
        getattr(tfm, effect_type)(**effect_params)
        y_out = tfm.build_array(input_array=y,
                                sample_rate_in=SR)
        return y_out
    return sox_effect