fix: remove references to nonexistent keys in vqgan dataset test (#1206)
The __main__ test block accessed batch["features"] and
batch["feature_lengths"], but VQGANCollator only returns "audios"
and "audio_lengths", causing a KeyError when running the file directly.
Made-with: Cursor