Updated register.py

Author: darikoneil

suite2p/registration/register.py

@@ -611,8 +611,9 @@ def registration_wrapper(f_reg, f_raw=None, f_reg_chan2=None, f_raw_chan2=None,
 
     outputs = compute_reference_and_register_frames(f_align_in, f_align_out=f_align_out, refImg=refImg, ops=ops)
     refImg, rmin, rmax, mean_img, rigid_offsets, nonrigid_offsets, zest = outputs
-    if len(rigid_offsets) > 1:
-            yoff, xoff, corrXY = rigid_offsets
+    yoff, xoff, corrXY = rigid_offsets
+
+            
     if ops['nonrigid']:
             yoff1, xoff1, corrXY1 = nonrigid_offsets
     else:
@@ -752,10 +753,9 @@ def save_registration_outputs_to_ops(registration_outputs, ops):
     ops['refImg'] = refImg 
     ops['rmin'], ops['rmax'] = rmin, rmax
     # assign rigid offsets to ops
-    if len(rigid_offsets>1):
-        ops['yoff'], ops['xoff'], ops['corrXY'] = rigid_offsets
+    ops['yoff'], ops['xoff'], ops['corrXY'] = rigid_offsets
     # assign nonrigid offsets to ops
-    if len(nonrigid_offsets>1):
+    if ops['nonrigid']:
         ops['yoff1'], ops['xoff1'], ops['corrXY1'] = nonrigid_offsets
     # assign mean images
     ops['meanImg'] = meanImg