chore: fix check code quality

Author: frascuchon

tests/features/test_midi.py

@@ -1,12 +1,9 @@
-import os
-import tempfile
 from unittest import TestCase
 
-import numpy as np
 import pytest
 
-from datasets import Dataset, Features, load_dataset
-from datasets.features import Midi, Value
+from datasets import Dataset, Features
+from datasets.features import Midi
 
 
 class TestMidiFeature(TestCase):
@@ -17,15 +14,15 @@ def test_audio_feature_type(self):
 
     def test_audio_feature_encode_example(self):
         midi = Midi()
-        
+
         # Test with path
         encoded = midi.encode_example("path/to/midi.mid")
         assert encoded == {"bytes": None, "path": "path/to/midi.mid"}
-        
+
         # Test with bytes
         encoded = midi.encode_example(b"fake_midi_bytes")
         assert encoded == {"bytes": b"fake_midi_bytes", "path": None}
-        
+
         # Test with dict containing notes
         notes_data = {
             "notes": [[60, 64, 0.0, 1.0], [62, 64, 1.0, 2.0]],
@@ -38,7 +35,7 @@ def test_audio_feature_encode_example(self):
 
     def test_audio_feature_decode_example(self):
         midi = Midi()
-        
+
         # Test decode with bytes
         fake_midi_bytes = b'MThd\x00\x00\x00\x06\x00\x01\x00\x02\x00\xdcMTrk\x00\x00\x00\x13\x00\xffQ\x03\x07\xa1 \x00\xffX\x04\x04\x02\x18\x08\x01\xff/\x00MTrk\x00\x00\x00\x16\x00\xc0\x00\x00\x90<@\x838<\x00\x00>@\x838>\x00\x01\xff/\x00'
         decoded = midi.decode_example({"bytes": fake_midi_bytes, "path": None})
@@ -50,11 +47,10 @@ def test_audio_feature_decode_example(self):
     def test_audio_feature_with_dataset(self):
         features = Features({"midi": Midi()})
         data = {"midi": ["fake_path1.mid", "fake_path2.mid"]}
-        
-        with tempfile.TemporaryDirectory() as tmp_dir:
-            dataset = Dataset.from_dict(data, features=features)
-            assert "midi" in dataset.column_names
-            assert dataset.features["midi"].dtype == "dict"
+
+        dataset = Dataset.from_dict(data, features=features)
+        assert "midi" in dataset.column_names
+        assert dataset.features["midi"].dtype == "dict"
 
     def test_audio_feature_decode_false(self):
         midi = Midi(decode=False)
@@ -68,10 +64,10 @@ def test_audio_feature_resolution(self):
     def test_audio_feature_flatten(self):
         midi = Midi(decode=False)
         flattened = midi.flatten()
-        assert "bytes" in flattened
-        assert "path" in flattened
+        assert "bytes" in flattened # type: ignore
+        assert "path" in flattened # type: ignore
 
     def test_audio_feature_decode_error(self):
         midi = Midi(decode=False)
         with pytest.raises(RuntimeError):
-            midi.decode_example({"bytes": b"fake", "path": None})
+            midi.decode_example({"bytes": b"fake", "path": None})