From dd047fa1f7589a8b4c77f1f5034417aba5a83662 Mon Sep 17 00:00:00 2001
From: Logan Adams <loadams@microsoft.com>
Date: Thu, 16 Jan 2025 13:18:45 -0800
Subject: [PATCH] Update import for torchvision.transformers

---
 tests/unit/alexnet_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/unit/alexnet_model.py b/tests/unit/alexnet_model.py
index 51e80e7f9e62..1ca5eedbcc86 100644
--- a/tests/unit/alexnet_model.py
+++ b/tests/unit/alexnet_model.py
@@ -84,7 +84,7 @@ def cast_to_half(x):
 
 def cifar_trainset(fp16=False):
     torchvision = pytest.importorskip("torchvision", minversion="0.5.0")
-    import torchvision.transforms as transforms
+    from torchvision import transformers
 
     transform_list = [
         transforms.ToTensor(),