fix single pil image handling

climategan_wrapper.py

@@ -169,6 +169,16 @@ class ClimateGAN:
             raise e
 
     def _preprocess_image(self, img):
+        """
+        Turns a HxWxC uint8 numpy array into a 640x640x3 float32 numpy array
+        in [-1, 1].
+
+        Args:
+            img (np.array): Image to resize crop and rescale
+
+        Returns:
+            np.array: Resized, cropped and rescaled image
+        """
         # rgba to rgb
         data = img if img.shape[-1] == 3 else uint8(rgba2rgb(img) * 255)
 
@@ -322,7 +332,7 @@ class ClimateGAN:
 
         if pil_image is not None:
             print("Warning: `pil_image` has been provided, it will override `images`")
-            images = self._preprocess_image(np.array(pil_image)[None, ...])
+            images = self._preprocess_image(np.array(pil_image))[None, ...]
 
         # Retrieve numpy events as a dict {event: array[BxHxWxC]}
         outputs = self.trainer.infer_all(