@@ -18,11 +18,16 @@ def check_import_si_full():
1818 import spikeinterface .full as si
1919
2020
21- def _create_recording ():
21+ def _create_recording (short = False ):
2222 import spikeinterface .full as si
2323
24+ if short :
25+ durations = [30.0 ]
26+ else :
27+ durations = [200.0 ]
28+
2429 rec , _ = si .generate_ground_truth_recording (
25- durations = [ 200.0 ] , sampling_frequency = 30_000.0 , num_channels = 16 , num_units = 10 , seed = 2205
30+ durations = durations , sampling_frequency = 30_000.0 , num_channels = 16 , num_units = 10 , seed = 2205
2631 )
2732 rec .save (folder = "./toy_example_recording" , verbose = False , ** job_kwargs )
2833
@@ -85,20 +90,24 @@ def _clean():
8590# add ci flag so that gui will not be used in ci
8691# end user can ignore
8792parser .add_argument ("--ci" , action = "store_false" )
93+ parser .add_argument ("--short" , action = "store_false" )
94+ parser .add_argument ("--skip-kilosort4" , action = "store_true" )
95+
8896
8997if __name__ == "__main__" :
9098
9199 args = parser .parse_args ()
92100
93101 _clean ()
94- _create_recording ()
102+ _create_recording (short = args . short )
95103
96104 steps = [
97105 ("Import spikeinterface" , check_import_si ),
98106 ("Import spikeinterface.full" , check_import_si_full ),
99107 ("Run tridesclous2" , run_tridesclous2 ),
100- ("Run kilosort4" , run_kilosort4 ),
101108 ]
109+ if not args .skip_kilosort4 :
110+ steps .append (("Run kilosort4" , run_kilosort4 ))
102111
103112 # backwards logic because default is True for end-user
104113 if args .ci :
0 commit comments