add new table to concat function

AllenInstitute · Dec 3, 2024 · d2a4716 · d2a4716
1 parent aa9bbb0
commit d2a4716
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/dynamic_routing_analysis/decoding_utils.py b/src/dynamic_routing_analysis/decoding_utils.py
@@ -2448,6 +2448,12 @@ def concat_decoder_summary_tables(dir,savepath):
         decoder_confidence_versus_trials_since_rewarded_target=pd.concat(decoder_confidence_versus_trials_since_rewarded_target_files)
         decoder_confidence_versus_trials_since_rewarded_target.to_pickle(upath.UPath(savepath) / ('decoder_confidence_versus_trials_since_rewarded_target_'+str(nu)+'_units.pkl'))
 
+        decoder_confidence_all_trials_files=[]
+        for p in dir.glob('*decoder_confidence_all_trials_'+str(nu)+'_units.pkl'):
+            decoder_confidence_all_trials_files.append(pd.read_pickle(p))
+        decoder_confidence_all_trials=pd.concat(decoder_confidence_all_trials_files)
+        decoder_confidence_all_trials.to_pickle(upath.UPath(savepath) / ('decoder_confidence_all_trials_'+str(nu)+'_units.pkl'))
+
         decoder_confidence_before_after_target_files=[]
         for p in dir.glob('*decoder_confidence_before_after_target_'+str(nu)+'_units.pkl'):
             decoder_confidence_before_after_target_files.append(pd.read_pickle(p))