depend on codex-africanus master

ratt-ru · Apr 15, 2024 · 5ddf08b · 5ddf08b
1 parent d083703
commit 5ddf08b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git 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
@@ -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"
             ]