From 021fac6defda204f14f813c9fb68da58bc71210b Mon Sep 17 00:00:00 2001
From: Chris Harris <cjh@lbl.gov>
Date: Wed, 16 Apr 2025 15:48:00 +0000
Subject: [PATCH] Fix TFDS example

Looks like the record set name has changed also with the split
names.
---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 5a50cb1dd..49978e038 100644
--- a/README.md
+++ b/README.md
@@ -42,12 +42,12 @@ print(mlc.Dataset(url).metadata.to_json())
 import tensorflow_datasets as tfds
 builder = tfds.core.dataset_builders.CroissantBuilder(
     jsonld=url,
-    record_set_ids=["record_set_fashion_mnist"],
+    record_set_ids=["fashion_mnist"],
     file_format='array_record',
 )
 builder.download_and_prepare()
 # 4. Split for training/testing
-train, test = builder.as_data_source(split=['default[:80%]', 'default[80%:]'])
+train, test = builder.as_data_source(split=['train', 'test'])
 ```
 
 Please see the [notebook recipes](python/mlcroissant/recipes) for more examples.