Fix script

Author: alejoe91

.github/workflows/generate_cambridge_neurotech_library.yml

@@ -29,7 +29,6 @@ jobs:
         run: |
           # probeinterface
           git clone https://github.com/spikeinterface/probeinterface ./probeinterface
-          ls probeinterface/resources
 
           # cambridge neurotech probe maps
           git clone https://github.com/cambridge-neurotech/probe_maps.git ./probe_maps
@@ -43,7 +42,7 @@ jobs:
           python ../probeinterface/resources/generate_cambridgeneurotech_library.py ../probe_maps/ --output-folder ./cambridge_library_generated/
 
           # check for json changes (except probeinterface version)
-          python check_for_json_changes_in_probes.py ../cambridgeneurotech/ ./cambridge_library_generated/
+          python check_for_json_changes_in_probes.py ../cambridgeneurotech/ ./cambridge_library_generated/ --copy-figures
           # clean up
           rm -r cambridge_library_generated/
           cd ..
@@ -62,7 +61,7 @@ jobs:
             echo "No changes to commit"
             echo "changes=false" >> $GITHUB_OUTPUT
           else
-            git commit -m "Update json files for NP probes"
+            git commit -m "Update json files for cambridge neurotech probes"
             echo "changes=true" >> $GITHUB_OUTPUT
           fi
 

scripts/check_for_json_changes_in_probes.py

@@ -17,6 +17,12 @@
     help="Path to the new probes directory",
 )
 
+parser.add_argument(
+    "--copy-figures",
+    action="store_true",
+    help="If set, copies figures as well when JSON files are different.",
+)
+
 
 if __name__ == "__main__":
     args = parser.parse_args()
@@ -40,5 +46,11 @@
             if lines1[3:] == lines2[3:]:
                 continue
             else:
-                shutil.copy(f"{temp_probe_json_path}", f"../imec/{probe_name}")
+                shutil.copy(f"{temp_probe_json_path}", old_dir / probe_name)
+                if args.copy_figures:
+                    temp_figure_path = temp_probe_directory / (probe_name + '.png')
+                    shutil.copy(f"{temp_figure_path}", old_dir / probe_name)
+                
+                
+