classy-szfast 0.0.25.post18__tar.gz → 0.0.25.post20__tar.gz

{classy_szfast-0.0.25.post18 → classy_szfast-0.0.25.post20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: classy_szfast
-Version: 0.0.25.post18
+Version: 0.0.25.post20
 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.post18 → classy_szfast-0.0.25.post20}/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
@@ -515,8 +525,8 @@ class classy_sz(classy):
         for product, collector in self.collectors.items():
             # print(product,collector)
             # Special case: sigma8 needs H0, which cannot be known beforehand:
-            if "sigma8" in self.collectors:
-                self.collectors["sigma8"].args[0] = 8 / self.classy.h()
+            # if "sigma8" in self.collectors:
+            #     self.collectors["sigma8"].args[0] = 8 / self.classy.h()
             method = getattr(self.classy, collector.method)
             arg_array = self.collectors[product].arg_array
             if isinstance(arg_array, int):

{classy_szfast-0.0.25.post18 → classy_szfast-0.0.25.post20}/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", False)
 import jax.numpy as jnp
 import jax.scipy as jscipy
 
@@ -173,7 +175,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.post18 → classy_szfast-0.0.25.post20}/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", False)
 import jax.numpy as jnp
 from .restore_nn import Restore_NN
 from .restore_nn import Restore_PCAplusNN

{classy_szfast-0.0.25.post18 → classy_szfast-0.0.25.post20}/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", False)
 import jax.numpy as jnp
 # import cosmopower
 # import classy_sz as csz

{classy_szfast-0.0.25.post18 → classy_szfast-0.0.25.post20}/classy_szfast.egg-info/PKG-INFO

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

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