diff --git a/pfb/workers/spotless.py b/pfb/workers/spotless.py index be0ad9636..f2f68de98 100644 --- a/pfb/workers/spotless.py +++ b/pfb/workers/spotless.py @@ -231,13 +231,6 @@ def _spotless(ddsi=None, **kw): rms_comps = np.std(np.sum(psiHoutvar, axis=0), axis=(-1,-2))[:, None, None] # preserve axes - # This is attempt at getting the primal-dual to converge - # more homogeneously when bands have very different wsums - # sfactor is used to scale the grad21 function below - sfactor = np.zeros((nband, 1, 1)) - sfactor[fsel] = wsum/wsums[fsel, None, None] # missing subbands - sfactor /= np.mean(sfactor[fsel]) - # TODO - load from dds if present if dual is None: dual = np.zeros((nband, nbasis, Nymax, Nxmax), dtype=dirty.dtype) diff --git a/setup.py b/setup.py index 7717c29d3..bd11be49d 100644 --- a/setup.py +++ b/setup.py @@ -20,7 +20,10 @@ "stimela >= 2.0rc14", "streamjoy", "tbb", - "jax[cpu]" + "jax[cpu]", + "codex-africanus[complete]" + "@git+https://github.com/ratt-ru/codex-africanus.git" + "@master" ]