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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: classy_szfast
-Version: 0.0.25.post21
+Version: 0.0.25.post23
 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.post21 → classy_szfast-0.0.25.post23}/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
 
@@ -715,6 +715,8 @@ class Class_szfast(object):
             self.cp_predicted_der = self.cp_der_nn[self.cosmo_model].ten_to_predictions_np(params_dict)[0]
         self.sigma8 = self.cp_predicted_der[1]
         self.Neff = self.cp_predicted_der[4]
+        self.ra_star = self.cp_predicted_der[12]
+
         return 0
 
 
@@ -839,7 +841,7 @@ class Class_szfast(object):
         pk_re = self.transpose(pk_re)
 
         
-        return pk_re, k_arr
+        return pk_re.flatten(), k_arr
     
 
     def calculate_pknl_at_z(self,
@@ -887,7 +889,7 @@ class Class_szfast(object):
         pk_re = self.transpose(pk_re)
 
         
-        return pk_re, k_arr
+        return pk_re.flatten(), k_arr
     
 
     def calculate_hubble(self,

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

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

@@ -3,7 +3,7 @@ requires = ["setuptools", "wheel"]
 build-backend = "setuptools.build_meta"
 
 [project]
-version = "0.0.25.post21"
+version = "0.0.25.post23"
 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]