classy-szfast 0.0.25.post19__tar.gz → 0.0.25.post21__tar.gz

{classy_szfast-0.0.25.post19 → classy_szfast-0.0.25.post21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: classy_szfast
-Version: 0.0.25.post19
+Version: 0.0.25.post21
 Summary: The accelerator of the class_sz code from https://github.com/CLASS-SZ
 Maintainer-email: Boris Bolliet <bb667@cam.ac.uk>
 License: MIT

{classy_szfast-0.0.25.post19 → classy_szfast-0.0.25.post21}/classy_szfast/classy_sz.py

@@ -7,6 +7,9 @@ import os
 import numpy as np
 import time
 
+import warnings
+
+
 class classy_sz(classy):
 
     use_class_sz_fast_mode = 1 # this is passed in the yaml file
@@ -22,7 +25,14 @@ class classy_sz(classy):
         if not self.classy_module:
             raise NotInstalledError(
                 self.log, "Could not find CLASS_SZ. Check error message above.")
-        
+
+        # Ignore the specific UserWarning from mcfit about using backend='jax'
+        warnings.filterwarnings(
+            "ignore",
+            message="use backend='jax' if desired",
+            category=UserWarning
+        )
+
         from classy_sz import Class, CosmoSevereError, CosmoComputationError
 
         # import classy_sz

{classy_szfast-0.0.25.post19 → classy_szfast-0.0.25.post21}/classy_szfast/classy_szfast.py

@@ -12,6 +12,8 @@ from multiprocessing import Process
 from mcfit import TophatVar
 from scipy.interpolate import CubicSpline
 import pickle
+import jax 
+jax.config.update("jax_enable_x64", True)
 import jax.numpy as jnp
 import jax.scipy as jscipy
 
@@ -129,7 +131,16 @@ class Class_szfast(object):
             self.pow = jnp.power
 
             self.sigma_B = 2. * self.pow(self.pi,5) * self.pow(Const._k_B_,4) / 15. / self.pow(Const._h_P_,3) / self.pow(Const._c_,2)
-
+            # print('sigma_B',self.sigma_B)
+            # print('pi',self.pi)
+            # print('k_B',Const._k_B_)
+            # print('h_P',Const._h_P_)
+            # print('c',Const._c_)
+            # print('pow(pi,5)',self.pow(self.pi,5)) 
+            # print('pow(k_B,4)',self.pow(Const._k_B_,4)) ## this doesnt work if float64 is disabled
+            # print('15.',15.)
+            # print('pow(h_P,3)',self.pow(Const._h_P_,3)) ## this doesnt work if float64 is disabled
+            # print('pow(c,2)',self.pow(Const._c_,2))
             self.linspace = jnp.linspace
             self.geomspace = jnp.geomspace
             self.arange = jnp.arange
@@ -173,7 +184,7 @@ class Class_szfast(object):
             self.cp_pkl_fftlog_alphas_real_nn  = cp_pkl_fftlog_alphas_real_nn
             self.cp_pkl_fftlog_alphas_imag_nn = cp_pkl_fftlog_alphas_imag_nn
 
-        self.cosmo_model = 'ede-v2'
+        self.cosmo_model = 'lcdm'
 
         self.use_Amod = 0
         self.Amod = 0 

{classy_szfast-0.0.25.post19 → classy_szfast-0.0.25.post21}/classy_szfast/cosmopower_jax.py

@@ -1,5 +1,7 @@
 from .config import path_to_class_sz_data
 import numpy as np
+import jax
+jax.config.update("jax_enable_x64", True)
 import jax.numpy as jnp
 from .restore_nn import Restore_NN
 from .restore_nn import Restore_PCAplusNN

{classy_szfast-0.0.25.post19 → classy_szfast-0.0.25.post21}/classy_szfast/utils.py

@@ -14,6 +14,7 @@ from numpy import linalg as LA
 import mcfit
 from mcfit import P2xi
 import jax
+jax.config.update("jax_enable_x64", True)
 import jax.numpy as jnp
 # import cosmopower
 # import classy_sz as csz

{classy_szfast-0.0.25.post19 → classy_szfast-0.0.25.post21}/classy_szfast.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: classy_szfast
-Version: 0.0.25.post19
+Version: 0.0.25.post21
 Summary: The accelerator of the class_sz code from https://github.com/CLASS-SZ
 Maintainer-email: Boris Bolliet <bb667@cam.ac.uk>
 License: MIT

{classy_szfast-0.0.25.post19 → classy_szfast-0.0.25.post21}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["setuptools", "wheel"]
 build-backend = "setuptools.build_meta"
 
 [project]
-version = "0.0.25.post19"
+version = "0.0.25.post21"
 license = { text = "MIT" }
 name = "classy_szfast"
 maintainers = [{name = "Boris Bolliet",email="bb667@cam.ac.uk"}]