Fix unbound local error in online_cnmf

flatironinstitute · Aug 9, 2024 · 61fff86 · 61fff86
1 parent cd61a86
commit 61fff86
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/caiman/source_extraction/cnmf/online_cnmf.py b/caiman/source_extraction/cnmf/online_cnmf.py
@@ -270,7 +270,6 @@ def _prepare_object(self, Yr, T, new_dims=None, idx_components=None):
 
         if self.is1p:
             estim = self.estimates
-            d1, d2 = estim.dims    
             estim.Yres_buf -= estim.b0
             if ssub_B == 1:
                 estim.Atb = estim.Ab.T.dot(estim.W.dot(estim.b0) - estim.b0)
@@ -389,6 +388,7 @@ def get_indices_of_pixels_on_ring(self, pixel):
                 else:
                     Yres -= estim.upscale_matrix.dot(estim.W.dot(
                         estim.downscale_matrix.dot(Yres)))
+            d1, d2 = self.estimates.dims
             Yres = Yres.reshape((d1, d2, -1), order='F')
 
             (self.estimates.first_moment, self.estimates.second_moment,