diff --git a/realbook/callbacks/spectrogram_visualization.py b/realbook/callbacks/spectrogram_visualization.py index fbca66c..e700a31 100644 --- a/realbook/callbacks/spectrogram_visualization.py +++ b/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" diff --git a/realbook/vendor/librosa_filters.py b/realbook/vendor/librosa_filters.py index 1a9bdff..381ec75 100644 --- a/realbook/vendor/librosa_filters.py +++ b/realbook/vendor/librosa_filters.py @@ -14,6 +14,9 @@ # TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE # OF THIS SOFTWARE. +# This is all copied from Librosa: don't bother type checking with MyPy. +# type: ignore + import warnings import scipy import numpy as np