minor bug updates

Author: lhparker1

downstream_tasks/similarity_search/embed_astroclip.py

@@ -64,12 +64,16 @@ def embed_astroclip(
                 .cpu()
                 .numpy()
             )
+            images.append(batch_test["image"])
+            spectra.append(batch_test["spectrum"])
 
     # Save as an HDF5 file
     with h5py.File(save_path, "w") as f:
         f.create_dataset("image_embeddings", data=np.concatenate(im_embeddings))
         f.create_dataset("spectrum_embeddings", data=np.concatenate(sp_embeddings))
         f.create_dataset("object_id", data=np.concatenate(obj_ids))
+        f.create_dataset("image", data=np.concatenate(images))
+        f.create_dataset("spectrum", data=np.concatenate(spectra))
     print(f"Embeddings saved to {save_path}")