fixed Pillow RGB convertion for transform_image function

Author: Dpbm

tests/test_utils.py

@@ -21,7 +21,7 @@ def image_path() -> str:
 @pytest.fixture()
 def image_obj(image_path) -> Image:
     """return a PIL image"""
-    return Image.open(image_path).convert("RGB")
+    return Image.open(image_path)
 
 
 @pytest.fixture()

utils/image.py

@@ -48,6 +48,7 @@ def transform_image(
     transform_level: TransformationLevel = TransformationLevel.ALL,
 ) -> torch.Tensor:
     """Transform and normalize a PIL image into a torch tensor ranging values from 0 to 1"""
+    img = img.convert("RGB")
 
     image_tensor = v2.Compose(
         get_transformations_pipeline(width, height, transform_level)