From 3f13ad3d79790d02136f4e3ebcd2c8b671615a23 Mon Sep 17 00:00:00 2001
From: Taylor Ninesling <taylor.ninesling@gmail.com>
Date: Mon, 3 Jun 2024 23:27:24 -0500
Subject: [PATCH] Fix dataset id for MNIST (#2238)

---
 candle-datasets/src/vision/mnist.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/candle-datasets/src/vision/mnist.rs b/candle-datasets/src/vision/mnist.rs
index 2dac883c8e..eb79e17e6f 100644
--- a/candle-datasets/src/vision/mnist.rs
+++ b/candle-datasets/src/vision/mnist.rs
@@ -89,7 +89,7 @@ fn load_parquet(parquet: SerializedFileReader<std::fs::File>) -> Result<(Tensor,
 
 pub fn load() -> Result<crate::vision::Dataset> {
     let api = Api::new().map_err(|e| Error::Msg(format!("Api error: {e}")))?;
-    let dataset_id = "mnist".to_string();
+    let dataset_id = "ylecun/mnist".to_string();
     let repo = Repo::with_revision(
         dataset_id,
         RepoType::Dataset,