From 4bd6641c5b1a7c1092e883b4217902c6f74d4e37 Mon Sep 17 00:00:00 2001 From: James Nightingale Date: Sat, 9 Mar 2024 18:35:18 +0000 Subject: [PATCH] comment out JAX --- autoarray/inversion/inversion/abstract.py | 1 - optional_requirements.txt | 4 +-- to_do_list | 44 ----------------------- 3 files changed, 2 insertions(+), 47 deletions(-) diff --git a/autoarray/inversion/inversion/abstract.py b/autoarray/inversion/inversion/abstract.py index 546d8b887..d637664b1 100644 --- a/autoarray/inversion/inversion/abstract.py +++ b/autoarray/inversion/inversion/abstract.py @@ -10,7 +10,6 @@ from autoarray.numba_util import profile_func from autoarray.inversion.linear_obj.linear_obj import LinearObj -from autoarray.inversion.linear_obj.func_list import AbstractLinearObjFuncList from autoarray.inversion.pixelization.mappers.abstract import AbstractMapper from autoarray.inversion.regularization.abstract import AbstractRegularization from autoarray.inversion.inversion.settings import SettingsInversion diff --git a/optional_requirements.txt b/optional_requirements.txt index 65383afef..72beef3c9 100644 --- a/optional_requirements.txt +++ b/optional_requirements.txt @@ -1,5 +1,5 @@ pylops>=1.10.0,<=1.18.3 pynufft -jax==0.4.3 -jaxlib==0.4.3 +#jax==0.4.3 +#jaxlib==0.4.3 #numba diff --git a/to_do_list b/to_do_list index 6596415f7..e69de29bb 100644 --- a/to_do_list +++ b/to_do_list @@ -1,44 +0,0 @@ - - -- Think very carefully about how Preloads and w_tilde interface with database. Pickle array? Large filesize -but fast. Need to be careful with galaxies. - - -__Refactor__ - -- Use composition in visualization MatPlot objects instead of inheritance. -- Extract geometry from mask package. -- auto_labels in visuals as a config loady thing. -- Split wrap / mat_wrap up into separate moduels. - -__Visualization__ - -- Need to properly make it so one can disable many aspections of visualization. - - -__Inversion With Multiple Multiwavelength Datasets__ - -Can create a list of InversionMatrix objects for each dataset + tracer and then have a new class which takes a list of -these objects and creates a combined curvature_reg matrix from them, which solves for all datasets simultaneously. - -This is genuine because it can reuse all existing functionality and doesn't require diting of the inner code -to get multi dataset fits working. - -The regularization matrix could addition addition multi-data regularization terms added. - -Multi analysis API can easily be used for model parameterization, with CombineAnalysis overwritten to define -the log likelihood fuinction (which maybe uses a special type of Fit object). - - - -__Probably Terrible Idea Notes__ - -w_tilde_tlc: - - - Faster calculation of w_tilde_imaging (or quick way to get from old one?). - - - docstrings in Preloads for w_tilde. - - - Proper unit test on Preloads for w_tilde. - - - Add options for implciit preload of w_tilde / rest on and off, and manual input of snr_cut? Most likely make it so \ No newline at end of file