Fix formatting and type checking for vendored code.

spotify · May 11, 2023 · a8406f7 · a8406f7
1 parent 7ac94ec
commit a8406f7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -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