daysago3

Author: miloswrath

code/data_processing/plot_utils.py

@@ -168,7 +168,6 @@ def _rt(df):
 
             return ax2
         filtered = filter(df)
-        print(filtered.head())
         acc = _percent_acc(filtered)
         rt = _rt(filtered)
 

code/main_handler.py

@@ -108,29 +108,38 @@ def qc_cc_dfs(self, dfs, task):
                                      task=task)
         return categories, plots
 
-
     def qc_ps_dfs(self, dfs, task):
         categories, plots = [], []
         plot_instance = PS_PLOTS()
         print(task)
+
         if task in ['PC', 'LC']:
             ps_instance = PS_QC('response_time', 'correct', 1, 0, 'block_c', 30000)
-            for df in dfs:
-                subject = df['subject_id'][1]
+            for df in dfs[:]:  # use a copy to safely modify the list during iteration
+                try:
+                    subject = df['subject_id'][1]
+                except Exception as e:
+                    print(f"Skipping df due to error accessing subject_id: {e}")
+                    dfs.remove(df)
+                    continue
+
                 print(f"qcing {subject}")
-                category = ps_instance.ps_qc(df, threshold=0.6,)
-                if task == 'PC':
-                    plot = plot_instance.lc_plot(df)
-                elif task == 'LC':
-                    plot = plot_instance.lc_plot(df)
+                category = ps_instance.ps_qc(df, threshold=0.6)
+                plot = plot_instance.lc_plot(df)
                 print(f"Category = {category}")
                 categories.append([subject, category, df])
                 plots.append([subject, plot])
 
         else:
             ps_instance = PS_QC('block_dur', 'correct', 1, 0, 'block_c', 125)
-            for df in dfs:
-                subject = df['subject_id'][1]
+            for df in dfs[:]:
+                try:
+                    subject = df['subject_id'][1]
+                except Exception as e:
+                    print(f"Skipping df due to error accessing subject_id: {e}")
+                    dfs.remove(df)
+                    continue
+
                 print(f"qcing {subject}")
                 category = ps_instance.ps_qc(df, threshold=0.6, DSST=True)
                 plot = plot_instance.dsst_plot(df)
@@ -139,10 +148,8 @@ def qc_ps_dfs(self, dfs, task):
                 plots.append([subject, plot])
 
         save_instance = SAVE_EVERYTHING()
-        save_instance.save_dfs(categories=categories,
-                                task=task)
-        save_instance.save_plots(plots=plots,
-                                    task=task)
+        save_instance.save_dfs(categories=categories, task=task)
+        save_instance.save_plots(plots=plots, task=task)
 
         return categories, plots