Update import for torchvision.transformers (#6958)

Fixes import - found via [torchfix](https://github.com/pytorch-labs/torchfix).
microsoft · Jan 17, 2025 · f97f088 · f97f088
1 parent 018ece5
commit f97f088
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 transforms
 
     transform_list = [
         transforms.ToTensor(),