From 10d1190b62dec978f76b42b56587456a1579c5e1 Mon Sep 17 00:00:00 2001 From: landmanbester Date: Sat, 30 Nov 2024 16:11:03 +0200 Subject: [PATCH] remove debug statements in clark clean --- pfb/deconv/clark.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/pfb/deconv/clark.py b/pfb/deconv/clark.py index 73c17adc..c748293d 100644 --- a/pfb/deconv/clark.py +++ b/pfb/deconv/clark.py @@ -8,7 +8,7 @@ import pyscilog log = pyscilog.get_logger('CLARK') -# @njit(nogil=True, cache=True) # parallel=True, +@njit(nogil=True, cache=True) # parallel=True, def subtract(A, psf, Ip, Iq, xhat, nxo2, nyo2): ''' Subtract psf centered at location of xhat @@ -25,7 +25,7 @@ def subtract(A, psf, Ip, Iq, xhat, nxo2, nyo2): return A -# @njit(nogil=True, cache=True) # parallel=True, +@njit(nogil=True, cache=True) # parallel=True, def subminor(A, psf, Ip, Iq, model, wsums, gamma=0.05, th=0.0, maxit=10000): """ Run subminor loop in active set @@ -83,9 +83,6 @@ def subminor(A, psf, Ip, Iq, model, wsums, gamma=0.05, th=0.0, maxit=10000): # mask = (left_x & right_x) & (left_y & right_y) mask = (np.abs(Idelp) <= nxo2) & (np.abs(Idelq) <= nyo2) - assert mask.size == Ip.size - assert mask.all() - # import ipdb; ipdb.set_trace() # Ipp, Iqq = psf[:, nxo2 - Ip[mask], nyo2 - Iq[mask]] A = subtract(A[:, mask], psf, Idelp[mask], Idelq[mask],