fix: processing on non-traditional file formats in blech_clust

Author: abuzarmahmood

.gitignore

@@ -6,3 +6,5 @@ requirements/.DS_Store
 blech.dir
 *.swp
 *.Rhistory
+.aider*
+.env

blech_clust.py

@@ -299,8 +299,6 @@ def generate_processing_scripts(dir_name, blech_clust_dir, electrode_layout_fram
 if file_type[0] != 'traditional':
     electrodes_list = file_lists[file_type[0]]['electrodes']
     dig_in_file_list = file_lists[file_type[0]]['dig_in']
-else:
-    rhd_file_list = file_lists[file_type[0]]['rhd']
 
     # Use info file for port list calculation
     info_file = np.fromfile(dir_name + '/info.rhd', dtype=np.dtype('float32'))
@@ -318,6 +316,7 @@ def generate_processing_scripts(dir_name, blech_clust_dir, electrode_layout_fram
     ports = info_dict['ports']
 
 if file_type[0] == 'traditional':
+    rhd_file_list = file_lists[file_type[0]]['rhd']
     with open(rhd_file_list[0], 'rb') as f:
         header = read_header(f)
     # temp_file, data_present = importrhdutilities.load_file(file_list[0])

pipeline_testing/prefect_pipeline.py

@@ -519,33 +519,29 @@ def emg_only_test():
 @flow(log_prints=True)
 def full_test():
     if break_bool:
-        for data_type in ['emg', 'emg_spike']:
-            print(f'Running full test with data type : {data_type}')
-            prep_data_flow(data_type = data_type)
-            run_spike_test()
-            run_emg_main_test()
+        spike_only_test()
+        emg_only_test()
+        spike_emg_test()
     else:
-        for data_type in ['emg', 'emg_spike']:
-            print(f'Running full test with data type : {data_type}')
-            try:
-                prep_data_flow(data_type = data_type)
-            except:
-                print('Failed to prep data')
-            try:
-                run_spike_test()
-            except:
-                print('Failed to run spike test')
-            try:
-                run_emg_main_test()
-            except:
-                print('Failed to run emg test')
+        try:
+            spike_only_test()
+        except:
+            print('Failed to run spike test')
+        try:
+            emg_only_test()
+        except:
+            print('Failed to run emg test')
+        try:
+            spike_emg_test()
+        except:
+            print('Failed to run spike+emg test')
 
 ############################################################
 ## Run Flows
 ############################################################
 # If no individual tests are required, run both
 if args.all: 
-    print('Running spike and emg tests')
+    print('Running all tests')
     full_test(return_state=True)
 elif args.e:
     print('Running emg tests only')