From 06fdb9fc0f9427ed488b2da988e23bdfa8df2d08 Mon Sep 17 00:00:00 2001
From: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>
Date: Fri, 20 Dec 2024 11:52:55 -0500
Subject: [PATCH] Propagate audio feature flag (#2633)

---
 crates/burn-core/Cargo.toml | 2 ++
 crates/burn/Cargo.toml      | 1 +
 2 files changed, 3 insertions(+)

diff --git a/crates/burn-core/Cargo.toml b/crates/burn-core/Cargo.toml
index 68b47b3826..e63af0fba5 100644
--- a/crates/burn-core/Cargo.toml
+++ b/crates/burn-core/Cargo.toml
@@ -38,6 +38,7 @@ doc = [
     "wgpu",
     "cuda-jit",
     "hip-jit",
+    "audio",
     "vision",
     "autodiff",
     "remote",
@@ -79,6 +80,7 @@ std = [
     "num-traits/std",
 ]
 vision = ["burn-dataset?/vision", "burn-common/network"]
+audio = ["burn-dataset?/audio"]
 
 # Backend
 autodiff = ["burn-autodiff"]
diff --git a/crates/burn/Cargo.toml b/crates/burn/Cargo.toml
index 583bfc1ddf..7f6af14fbb 100644
--- a/crates/burn/Cargo.toml
+++ b/crates/burn/Cargo.toml
@@ -32,6 +32,7 @@ dataset = ["burn-core/dataset"]
 sqlite = ["burn-core/sqlite"]
 sqlite-bundled = ["burn-core/sqlite-bundled"]
 
+audio = ["burn-core/audio"]
 vision = ["burn-core/vision"]
 
 # Backends