diff --git a/docs/source/how-to/extending/custom-scripts.md b/docs/source/how-to/extending/custom-scripts.md index 8e8961a5b6..5e78149fe1 100644 --- a/docs/source/how-to/extending/custom-scripts.md +++ b/docs/source/how-to/extending/custom-scripts.md @@ -36,7 +36,7 @@ class MyDataLoader: @Registry.register_dataloader() def my_dataloader(dataset, batch_size): - return MyDataloader(dataset, batch_size) + return MyDataLoader(dataset, batch_size) @Registry.register_post_process() def my_post_process(output):