Add separate weight paths for the two tracers if they have the same name

Author: andreicuceu

vega/correlation_item.py

@@ -48,8 +48,10 @@ def __init__(self, config, model_pk=False):
 
         self.new_metals = config['model'].getboolean('new_metals', False)
         if self.new_metals:
-            self.tracer1['delta-stack'] = config['data'].get(f'delta_stack_{self.tracer1["name"]}')
-            self.tracer2['delta-stack'] = config['data'].get(f'delta_stack_{self.tracer2["name"]}')
+            self.tracer1['weights-path'] = config['data'].get('weights-tracer1')
+            self.tracer2['weights-path'] = config['data'].get('weights-tracer2', None)
+            if self.tracer2['weights-path'] is None:
+                self.tracer2['weights-path'] = self.tracer1['weights-path']
 
         self.test_flag = config['data'].getboolean('test', False)
 

vega/metals.py

@@ -74,14 +74,11 @@ def __init__(self, corr_item, fiducial, scale_params, PktoXi_obj, data=None):
                 blinding='none'
             )
 
-            with fits.open(self._corr_item.tracer1['delta-stack']) as hdul:
+            with fits.open(self._corr_item.tracer1['weights-path']) as hdul:
                 self.stack_table_1 = hdul[1].data
 
-            if self._corr_item.tracer1['name'] == self._corr_item.tracer2['name']:
-                self.stack_table_2 = self.stack_table_1
-            else:
-                with fits.open(self._corr_item.tracer2['delta-stack']) as hdul:
-                    self.stack_table_2 = hdul[1].data
+            with fits.open(self._corr_item.tracer2['weights-path']) as hdul:
+                self.stack_table_2 = hdul[1].data
 
         # Initialize metals
         self.Pk_metal = None