[simulations] Save signal_label in crs

astrotools/simulations.py

@@ -110,6 +110,7 @@ class BaseSimulation:
         sets_ids_3d = np.repeat(np.arange(3), np.prod(self.shape)).reshape((3,) + self.shape)
         self.crs['vecs'] = self.crs['vecs'][sets_ids_3d, sets_ids, np.stack([shuffle_ids] * 3)]
         self.signal_label = self.signal_label[sets_ids, shuffle_ids]
+        self.crs['signal_label'] = self.signal_label
 
 
 class ObservedBound(BaseSimulation):