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

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

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

@@ -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.post22}/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.post22
 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.post21 → classy_szfast-0.0.25.post22}/pyproject.toml

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