From 0506c0a1fb8b176a09e364f1524c05206d48da35 Mon Sep 17 00:00:00 2001
From: "Thomson, Alec (CASS, Kensington)" <alec.thomson@csiro.au>
Date: Wed, 24 Jul 2024 09:43:55 +1000
Subject: [PATCH] Catch bad data

---
 arrakis/validate.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/arrakis/validate.py b/arrakis/validate.py
index a508497..94909eb 100644
--- a/arrakis/validate.py
+++ b/arrakis/validate.py
@@ -248,7 +248,7 @@ def plot_leakage(
                 np.full((bins, bins), np.nan),
                 np.linspace(-4, 4, bins),
                 np.linspace(-4, 4, bins),
-                WCS(),
+                None,
             )
 
     per_subplot_kw = {
@@ -267,6 +267,8 @@ def plot_leakage(
         sharey=True,
     )
     for stokes, ax in ax_dict.items():
+        if leakage_dict[stokes].wcs is None:
+            continue
         data = leakage_dict[stokes].data
         xc = leakage_dict[stokes].xc
         yc = leakage_dict[stokes].yc