Update import for torchvision.transformers (deepspeedai#6958)

Fixes import - found via [torchfix](https://github.com/pytorch-labs/torchfix). Signed-off-by: yisheng <yi.sheng@intel.com>
ys950902 · Mar 6, 2025 · ec940ef · ec940ef
1 parent e5d40a7
commit ec940ef
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(),