From 3a8dcfb5f4b1be7d23607f995390a308321e0105 Mon Sep 17 00:00:00 2001
From: nathaniel <nathaniel.simard.42@gmail.com>
Date: Fri, 1 Dec 2023 16:01:22 -0500
Subject: [PATCH] Fix burn-train feature flag

---
 burn-train/Cargo.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/burn-train/Cargo.toml b/burn-train/Cargo.toml
index 2cae34ebca..e635236cea 100644
--- 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