From 72b01eae5f82db78e301d9b15f5065b9e53f961f Mon Sep 17 00:00:00 2001
From: James Nightingale <james.w.nightingale@durham.ac.uk>
Date: Wed, 9 Apr 2025 09:26:43 +0100
Subject: [PATCH] converts JAX arrays for .fits output

---
 autoconf/fitsable.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/autoconf/fitsable.py b/autoconf/fitsable.py
index 9901630..fae5784 100644
--- a/autoconf/fitsable.py
+++ b/autoconf/fitsable.py
@@ -109,6 +109,12 @@ def hdu_list_for_output_from(
         if ext_name_list is not None:
             header["EXTNAME"] = ext_name_list[i].upper()
 
+        # Convert from JAX
+        try:
+            values = np.array(values.array)
+        except AttributeError:
+            values = np.array(values)
+
         values = flip_for_ds9_from(values)
                 
         if i == 0: