Fix burn-train feature flag

tracel-ai · Dec 1, 2023 · 3a8dcfb · 3a8dcfb
1 parent 4d79449
commit 3a8dcfb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/burn-train/Cargo.toml b/burn-train/Cargo.toml
@@ -16,7 +16,7 @@ metrics = ["nvml-wrapper", "sysinfo", "systemstat"]
 tui = ["ratatui", "crossterm"]
 
 [dependencies]
-burn-core = { path = "../burn-core", version = "0.11.0" }
+burn-core = { path = "../burn-core", version = "0.11.0", features = ["dataset"] }
 
 log = { workspace = true }
 tracing-subscriber.workspace = true