Resolve comments [3]

Author: Amar3tto

sdks/python/apache_beam/examples/inference/pytorch_image_captioning.py

@@ -202,8 +202,9 @@ def run_inference(
       uris.append(x.get("uri", ""))
       try:
         images.append(decode_pil(b))
-      except Exception:
+      except Exception as e:
         # fallback: a blank image (so pipeline keeps going)
+        logging.warning("Failed to decode image %s: %s", uris[-1], e)
         images.append(PILImage.new("RGB", (224, 224), color=(0, 0, 0)))
 
     # Processor makes pixel_values
@@ -305,7 +306,8 @@ def run_inference(
 
       try:
         img = decode_pil(image_bytes)
-      except Exception:
+      except Exception as e:
+        logging.warning("Failed to decode image for CLIP ranking: %s", e)
         img = PILImage.new("RGB", (224, 224), color=(0, 0, 0))
 
       start_i = len(texts)

sdks/python/apache_beam/examples/inference/pytorch_imagenet_rightfit.py

@@ -181,8 +181,12 @@ def process(self, kv: Tuple[str, PredictionResult]):
         # fallback: try first value if dict shape differs
         try:
           logits = next(iter(inference_obj.values()))
+          logging.warning(
+              'Could not find <logits> key in model output.'
+              'Falling back to first value in dict.'
+          )
         except Exception:
-          logits = None
+          logging.warning('Could not find <logits> key in dict.')
     else:
       logits = inference_obj
 
@@ -454,7 +458,7 @@ def run(
         mdl(torch.unsqueeze(dummy, 0))
       bs_ok = bs if bs is not None else 64
       break
-    except Exception as e:
+    except RuntimeError as e:
       last_err = e
       logging.warning("Batch size %s failed during warmup: %s", bs, e)
       continue