From 4eee3071be53c6ac7aa18b855e36b7f3c8515d7d Mon Sep 17 00:00:00 2001
From: chaitanya <69865263+ChaitanyaSirivuri@users.noreply.github.com>
Date: Tue, 8 Apr 2025 21:37:21 -0400
Subject: [PATCH] update data_iter.next()

Updatad data_iter.next() to next(data_iter).
---
 tutorials/01-basics/pytorch_basics/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tutorials/01-basics/pytorch_basics/main.py b/tutorials/01-basics/pytorch_basics/main.py
index 744400c2..c4f169a8 100644
--- a/tutorials/01-basics/pytorch_basics/main.py
+++ b/tutorials/01-basics/pytorch_basics/main.py
@@ -121,7 +121,7 @@
 data_iter = iter(train_loader)
 
 # Mini-batch images and labels.
-images, labels = data_iter.next()
+images, labels = next(data_iter)
 
 # Actual usage of the data loader is as below.
 for images, labels in train_loader: