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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: classy_szfast
-Version: 0.0.25.post22
+Version: 0.0.25.post24
 Summary: The accelerator of the class_sz code from https://github.com/CLASS-SZ
 Maintainer-email: Boris Bolliet <bb667@cam.ac.uk>
 License: MIT
@@ -13,4 +13,4 @@ Requires-Dist: tensorflow
 Requires-Dist: mcfit>=0.0.22
 Requires-Dist: get_cosmopower_emus>=0.0.15
 Requires-Dist: class_sz_data>=0.0.13
-Requires-Dist: cosmopower-jax
+Requires-Dist: cosmopower_jax==0.5.5

{classy_szfast-0.0.25.post22 → classy_szfast-0.0.25.post24}/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", True)
+jax.config.update("jax_enable_x64", False)
 import jax.numpy as jnp
 import jax.scipy as jscipy
 
@@ -744,7 +744,7 @@ class Class_szfast(object):
                                                     kind='linear',
                                                     axis=-1,
                                                     copy=True,
-                                                    bounds_error=None,
+                                                    bounds_error=False,
                                                     fill_value=np.nan,
                                                     assume_sorted=False)
 
@@ -935,8 +935,8 @@ class Class_szfast(object):
                                                 kind='linear',
                                                 axis=-1,
                                                 copy=True,
-                                                bounds_error=None,
-                                                fill_value=np.nan,
+                                                bounds_error=False,
+                                                fill_value=0.,
                                                 assume_sorted=False)
 
     def calculate_chi(self,
@@ -981,15 +981,15 @@ class Class_szfast(object):
             
                 self.cp_predicted_da  = self.cp_da_nn[self.cosmo_model].predictions_np(params_dict)[0]
             
-
+            
             self.chi_interp = scipy.interpolate.interp1d(
                                                         self.cp_z_interp,
                                                         self.cp_predicted_da*(1.+self.cp_z_interp),
                                                         kind='linear',
                                                         axis=-1,
                                                         copy=True,
-                                                        bounds_error=None,
-                                                        fill_value=np.nan,
+                                                        bounds_error=False,
+                                                        fill_value=0,
                                                         assume_sorted=False)
 
     def get_cmb_cls(self,ell_factor=True,Tcmb_uk = Tcmb_uk):

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: classy_szfast
-Version: 0.0.25.post22
+Version: 0.0.25.post24
 Summary: The accelerator of the class_sz code from https://github.com/CLASS-SZ
 Maintainer-email: Boris Bolliet <bb667@cam.ac.uk>
 License: MIT
@@ -13,4 +13,4 @@ Requires-Dist: tensorflow
 Requires-Dist: mcfit>=0.0.22
 Requires-Dist: get_cosmopower_emus>=0.0.15
 Requires-Dist: class_sz_data>=0.0.13
-Requires-Dist: cosmopower-jax
+Requires-Dist: cosmopower_jax==0.5.5

{classy_szfast-0.0.25.post22 → classy_szfast-0.0.25.post24}/classy_szfast.egg-info/requires.txt

@@ -4,4 +4,4 @@ tensorflow
 mcfit>=0.0.22
 get_cosmopower_emus>=0.0.15
 class_sz_data>=0.0.13
-cosmopower-jax
+cosmopower_jax==0.5.5

{classy_szfast-0.0.25.post22 → classy_szfast-0.0.25.post24}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["setuptools", "wheel"]
 build-backend = "setuptools.build_meta"
 
 [project]
-version = "0.0.25.post22"
+version = "0.0.25.post24"
 license = { text = "MIT" }
 name = "classy_szfast"
 maintainers = [{name = "Boris Bolliet",email="bb667@cam.ac.uk"}]
@@ -16,7 +16,7 @@ dependencies = [
     "mcfit>=0.0.22",
     "get_cosmopower_emus>=0.0.15",
     "class_sz_data>=0.0.13",
-    "cosmopower-jax"
+    "cosmopower_jax==0.5.5"
 ]
 
 [project.urls]