🩹 Fix absolute imports #2

CMBSciPol · May 11, 2024 · 9f03e7a · 9f03e7a
1 parent ddefcd5
commit 9f03e7a
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 13 deletions.
diff --git a/micmac/Harmonic_Likelihood_sampling.py b/micmac/Harmonic_Likelihood_sampling.py
@@ -28,18 +28,21 @@
 import toml
 from jax import config
 
-from .jax_Sampling_toolbox import (
+from micmac.jax_Sampling_toolbox import (
     Sampling_functions,
     multivariate_Metropolis_Hasting_step_numpyro_bounded,
 )
-from .jax_tools import (
+from micmac.jax_tools import (
     frequency_alms_x_obj_red_covariance_cell_JAX,
     get_reduced_matrix_from_c_ell_jax,
 )
-from .noisecovar import get_true_Cl_noise
-from .templates_spv import get_nodes_b, tree_spv_config
-from .tools import get_c_ells_from_red_covariance_matrix, get_reduced_matrix_from_c_ell
-from .utils import generate_power_spectra_CAMB
+from micmac.noisecovar import get_true_Cl_noise
+from micmac.templates_spv import get_nodes_b, tree_spv_config
+from micmac.tools import (
+    get_c_ells_from_red_covariance_matrix,
+    get_reduced_matrix_from_c_ell,
+)
+from micmac.utils import generate_power_spectra_CAMB
 
 config.update('jax_enable_x64', True)
 

diff --git a/micmac/Non_param_Likelihood_sampling.py b/micmac/Non_param_Likelihood_sampling.py
@@ -28,27 +28,28 @@
 from jax import config
 from jax_tqdm import scan_tqdm
 
-from .jax_Sampling_toolbox import (
+from micmac.jax_Sampling_toolbox import (
     Sampling_functions,
     separate_single_MH_step_index_accelerated,
     separate_single_MH_step_index_v2b,
     separate_single_MH_step_index_v4_pixel,
     separate_single_MH_step_index_v4b_pixel,
     single_Metropolis_Hasting_step,
 )
-from .jax_tools import (
+from micmac.jax_tools import (
     get_cell_from_map_jax,
     get_reduced_matrix_from_c_ell_jax,
     get_sqrt_reduced_matrix_from_matrix_jax,
     maps_x_red_covariance_cell_JAX,
 )
-from .noisecovar import (
+from micmac.noisecovar import (
     get_BtinvN,
     get_inv_BtinvNB,
     get_inv_BtinvNB_c_ell,
     get_noise_covar_extended,
     get_Wd,
 )
+
 from .templates_spv import get_nodes_b, tree_spv_config
 from .tools import get_c_ells_from_red_covariance_matrix, get_reduced_matrix_from_c_ell
 from .utils import generate_power_spectra_CAMB, get_instr

diff --git a/micmac/jax_Sampling_toolbox.py b/micmac/jax_Sampling_toolbox.py
@@ -24,7 +24,7 @@
 import numpyro.distributions as dist
 from jax import random
 
-from .jax_tools import (
+from micmac.jax_tools import (
     alm_dot_product_JAX,
     alms_x_red_covariance_cell_JAX,
     frequency_alms_x_obj_red_covariance_cell_JAX,
@@ -33,8 +33,8 @@
     get_sqrt_reduced_matrix_from_matrix_jax,
     maps_x_red_covariance_cell_JAX,
 )
-from .mixingmatrix import MixingMatrix
-from .noisecovar import get_BtinvN, get_inv_BtinvNB, get_inv_BtinvNB_c_ell
+from micmac.mixingmatrix import MixingMatrix
+from micmac.noisecovar import get_BtinvN, get_inv_BtinvNB, get_inv_BtinvNB_c_ell
 
 
 class Sampling_functions(MixingMatrix):

diff --git a/micmac/mixingmatrix.py b/micmac/mixingmatrix.py
@@ -21,7 +21,7 @@
 import jax.numpy as jnp
 import numpy as np
 
-from .templates_spv import (
+from micmac.templates_spv import (
     create_one_template,
     create_one_template_from_bdefaultvalue,
     get_n_patches_b,