From 6a9bbbc88673d4d7a29c9c463d92a4028651fc58 Mon Sep 17 00:00:00 2001
From: Teddy Koker <teddy.koker@gmail.com>
Date: Tue, 24 Mar 2020 14:53:15 -0400
Subject: [PATCH] Fix dataloaders in TPU example (#1174)

---
 docs/source/introduction_guide.rst | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/source/introduction_guide.rst b/docs/source/introduction_guide.rst
index 8d7ca49adb..c045383951 100644
--- a/docs/source/introduction_guide.rst
+++ b/docs/source/introduction_guide.rst
@@ -547,10 +547,10 @@ In this method we do all the preparation we need to do once (instead of on every
         return DataLoader(self.train_dataset, batch_size=64)
 
       def val_dataloader(self):
-        return DataLoader(self.mnist_val, batch_size=64)
+        return DataLoader(self.val_dataset, batch_size=64)
 
       def test_dataloader(self):
-        return DataLoader(self.mnist_test, batch_size=64)
+        return DataLoader(self.test_dataset, batch_size=64)
 
 The `prepare_data` method is also a good place to do any data processing that needs to be done only
 once (ie: download or tokenize, etc...).