Update zalign.py

Made sure that if a reg_file is given to the zalign function it uses it, rather than the ops reg_file
MouseLand · Mar 7, 2024 · 1a61fdb · 1a61fdb
1 parent 24a7f26
commit 1a61fdb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/suite2p/registration/zalign.py b/suite2p/registration/zalign.py
@@ -157,10 +157,10 @@ def compute_zpos(Zreg, ops, reg_file=None):
         ]
 
     reg_file = ops["reg_file"] if reg_file is None else reg_file
-    nbytes = os.path.getsize(ops["reg_file"])
+    nbytes = os.path.getsize(reg_file)
     nFrames = int(nbytes / (2 * Ly * Lx))
 
-    reg_file = open(ops["reg_file"], "rb")
+    reg_file = open(reg_file, "rb")
     refAndMasks = []
     for Z in Zreg:
         if ops["1Preg"]: