Vendor relevant Librosa filter code to avoid runtime dependencies on Numba and libsndfile.

realbook/callbacks/spectrogram_visualization.py

@@ -106,7 +106,7 @@ def on_train_begin(self, logs: Any = None) -> None:
 
             with self.tensorboard_writer.as_default():
                 # Pull n random batches from the dataset and send them to TensorBoard.
-                for (data, _) in self.example_batches:
+                for data, _ in self.example_batches:
                     assert tf.rank(data) == 2, "Expected input data to be of rank 2, with shape (batch, audio)."
                     assert tf.shape(data)[0] < tf.shape(data)[1], (
                         "Expected input data to be of rank 2, with shape (batch, audio), but got shape"

realbook/layers/signal.py

@@ -18,11 +18,11 @@
 import warnings
 from typing import Any, Callable, Dict, Optional, Union
 
-import librosa
 import tensorflow as tf
 import numpy as np
 
 from realbook.layers.math import log_base_b
+from realbook.vendor import librosa_filters
 
 
 def _create_padded_window(
@@ -209,7 +209,7 @@ def build(self, input_shape: tf.TensorShape) -> None:
             self.fft_length
         )  # type: ignore
 
-        self.window_sum = librosa.filters.window_sumsquare(
+        self.window_sum = librosa_filters.window_sumsquare(
             window=self.window.numpy(),
             n_frames=input_shape[0] if input_shape.rank == 2 else input_shape[1],
             win_length=self.window_length,
@@ -353,7 +353,7 @@ def build(self, input_shape: tf.TensorShape) -> None:
         super().build(input_shape)
 
         self.mel_weight_matrix = tf.constant(
-            librosa.filters.mel(
+            librosa_filters.mel(
                 sr=self.sample_rate,
                 n_fft=self.fft_length,
                 n_mels=self.n_mels,

realbook/vendor/__init__.py

@@ -0,0 +1,16 @@
+#!/usr/bin/env python
+# encoding: utf-8
+#
+# Copyright 2023 Spotify AB
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.