Fix trigger_idx call

smoia · smoia · commit 6571e6bc9292 · 2022-12-15T11:48:28.000+01:00
diff --git a/phys2bids/phys2bids.py b/phys2bids/phys2bids.py
@@ -294,7 +294,7 @@ def phys2bids(filename, info=False, indir='.', outdir='.', heur_file=None,
                                       os.path.splitext(os.path.basename(filename))[0])
             for i, take in enumerate(phys_in.keys()):
                 plot_fileprefix = f'{fileprefix}_{take:02d}'
-                viz.export_trigger_plot(phys_in[take], phys_in.trigger_idx, plot_fileprefix, tr[i],
+                viz.export_trigger_plot(phys_in[take], phys_in[take].trigger_idx, plot_fileprefix, tr[i],
                                         num_timepoints_expected[i], filename,
                                         sub, ses)
 
@@ -388,12 +388,12 @@ def phys2bids(filename, info=False, indir='.', outdir='.', heur_file=None,
                                                 phys_in[take].timeseries[0][-1],
                                                 num=phys_out[key].timeseries[0].shape[0]))
             # Add trigger channel in the proper frequency.
-            if uniq_freq != phys_in[take].freq[phys_in.trigger_idx]:
-                phys_out[key].ch_name.insert(1, phys_in[take].ch_name[phys_in.trigger_idx])
-                phys_out[key].units.insert(1, phys_in[take].units[phys_in.trigger_idx])
+            if uniq_freq != phys_in[take].freq[phys_in[take].trigger_idx]:
+                phys_out[key].ch_name.insert(1, phys_in[take].ch_name[phys_in[take].trigger_idx])
+                phys_out[key].units.insert(1, phys_in[take].units[phys_in[take].trigger_idx])
                 phys_out[key].timeseries.insert(1, np.interp(phys_out[key].timeseries[0],
                                                              phys_in[take].timeseries[0],
-                                                             phys_in[take].timeseries[phys_in.trigger_idx]))
+                                                             phys_in[take].timeseries[phys_in[take].trigger_idx]))
             phys_out[key] = BlueprintOutput.init_from_blueprint(phys_out[key])
 
         # Preparing output parameters: name and folder.
