classy-szfast 0.0.25.post24__tar.gz → 0.0.25.post26__tar.gz

{classy_szfast-0.0.25.post24 → classy_szfast-0.0.25.post26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: classy_szfast
-Version: 0.0.25.post24
+Version: 0.0.25.post26
 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.post24 → classy_szfast-0.0.25.post26}/classy_szfast/__init__.py

@@ -7,6 +7,8 @@ from .pks_and_sigmas import *
 from pathlib import Path
 import sys
 
+from .utils import Const
+
 
 from .custom_profiles.custom_profiles import *
 

{classy_szfast-0.0.25.post24 → classy_szfast-0.0.25.post26}/classy_szfast/classy_szfast.py

@@ -13,7 +13,7 @@ from mcfit import TophatVar
 from scipy.interpolate import CubicSpline
 import pickle
 import jax 
-jax.config.update("jax_enable_x64", False)
+jax.config.update("jax_enable_x64", True)
 import jax.numpy as jnp
 import jax.scipy as jscipy
 
@@ -130,12 +130,13 @@ class Class_szfast(object):
             self.log = jnp.log
             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)
+            # 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)
+            self.sigma_B = 5.6704004737209545e-08
             # 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('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.)
@@ -264,7 +265,9 @@ class Class_szfast(object):
             self.pk_power_fac= (dls)**-1
 
 
-        self.cp_z_interp = self.linspace(0.,20.,5000)
+        
+        self.cp_z_interp_zmax = 20.
+        self.cp_z_interp = self.linspace(0.,self.cp_z_interp_zmax,5000)
 
         self.csz_base = None
 

{classy_szfast-0.0.25.post24 → classy_szfast-0.0.25.post26}/classy_szfast/utils.py

@@ -60,7 +60,7 @@ z_recombination_max = 1500
 # ------------------
 # Light speed
 class Const:
-    c_km_s = 299792.458  # speed of light
+    c_km_s = 299792.458  # speed of light in km/s
     h_J_s = 6.626070040e-34  # Planck's constant
     kB_J_K = 1.38064852e-23  # Boltzmann constant
 

{classy_szfast-0.0.25.post24 → classy_szfast-0.0.25.post26}/classy_szfast.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: classy_szfast
-Version: 0.0.25.post24
+Version: 0.0.25.post26
 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.post24 → classy_szfast-0.0.25.post26}/pyproject.toml

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