Update KS2.5 and KS3 function

Author: alejoe91

kilosort_no_license/kilosort2_5-compiled/ks2_5_compiled.m

@@ -13,11 +13,16 @@ function ks2_5_compiled(fpath)
 
         % NEW STEP TO DO DATA REGISTRATION
         if isfield(ops, 'do_correction')
-            fprintf("Drift correction ENABLED")
             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

kilosort_no_license/kilosort3-compiled/ks3_compiled.m

@@ -13,11 +13,16 @@ function ks3_compiled(fpath)
 
         % run data registration
         if isfield(ops, 'do_correction')
-            fprintf("Drift correction ENABLED")
             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);