Merge remote-tracking branch 'upstream/main' into main

Author: chyumin

kilosort_no_license/kilosort2_5-compiled/ks2_5_compiled.m

@@ -12,7 +12,18 @@ function ks2_5_compiled(fpath)
         rez = preprocessDataSub(ops);
 
         % NEW STEP TO DO DATA REGISTRATION
-        rez = datashift2(rez, 1); % last input is for shifting data
+        if isfield(ops, 'do_correction')
+            do_correction = ops.do_correction;
+        else 
+            do_correction = 1;
+        end
+
+        if do_correction
+            fprintf("Drift correction ENABLED\n");
+        else
+            fprintf("Drift correction DISABLED\n");
+        end
+        rez = datashift2(rez, do_correction); % last input is for shifting data
 
         % ORDER OF BATCHES IS NOW RANDOM, controlled by random number generator
         iseed = 1;

kilosort_no_license/kilosort3-compiled/ks3_compiled.m

@@ -12,7 +12,18 @@ function ks3_compiled(fpath)
         rez = preprocessDataSub(ops);
 
         % run data registration
-        rez = datashift2(rez, 1); % last input is for shifting data
+        if isfield(ops, 'do_correction')
+            do_correction = ops.do_correction;
+        else 
+            do_correction = 1;
+        end
+
+        if do_correction
+            fprintf("Drift correction ENABLED\n");
+        else
+            fprintf("Drift correction DISABLED\n");
+        end
+        rez = datashift2(rez, do_correction); % last input is for shifting data
 
         [rez, st3, tF] = extract_spikes(rez);