Update test/test_pipeline/components/training/test_image_data_loader.py

Co-authored-by: Ravin Kohli <[email protected]>

Author: franchuterivera

test/test_pipeline/components/training/test_image_data_loader.py

@@ -24,7 +24,7 @@ def test_imageloader_build_transform():
 
     assert isinstance(compose, torchvision.transforms.Compose)
 
-    # We expect to tensor and image augmenter
+    # We expect there to be two transforms, to tensor and image augmenter
     assert len(compose.transforms) == 2
 
     compose = loader.build_transform(fit_dictionary, mode='test')