classy-szfast 0.0.3__tar.gz

classy_szfast-0.0.3/.gitignore

@@ -0,0 +1,7 @@
+
+.DS_Store
+*egg-info
+test_classy_szfast.py
+test_classy_szfast-mnu3state.py 
+__pycache__
+requirements.txt

classy_szfast-0.0.3/MANIFEST.in

@@ -0,0 +1 @@
+

classy_szfast-0.0.3/PKG-INFO

@@ -0,0 +1,16 @@
+Metadata-Version: 2.1
+Name: classy_szfast
+Version: 0.0.3
+Summary: The accelerator of the class_sz code from https://github.com/CLASS-SZ
+Home-page: https://github.com/CLASS-SZ/classy_szfast
+Download-URL: https://github.com/CLASS-SZ/classy_szfast
+Author: Boris Bolliet, Ola Kusiak
+Author-email: bb667@cam.ac.uk, akk2175@columbia.edu
+Maintainer-email: Boris Bolliet <bb667@cam.ac.uk>
+Project-URL: Homepage, https://github.com/CLASS-SZ
+Project-URL: GitHub, https://github.com/CLASS-SZ
+Description-Content-Type: text/markdown
+Requires-Dist: tensorflow==2.13.0
+Requires-Dist: tensorflow-probability==0.21.0
+Requires-Dist: cosmopower
+Requires-Dist: mcfit

classy_szfast-0.0.3/README.md

@@ -0,0 +1 @@
+Python module for the acceleration of class_sz and interface with likelihood codes. 

classy_szfast-0.0.3/classy_szfast/__init__.py

@@ -0,0 +1,10 @@
+from classy_szfast.classy_szfast import Class_szfast
+from .config import *
+from .utils import *
+from .cosmopower import *
+from .pks_and_sigmas import *
+
+from pathlib import Path
+import sys
+from .custom_profiles import *
+from .custom_bias import *

classy_szfast-0.0.3/classy_szfast/classy_sz.py

@@ -0,0 +1,585 @@
+from cobaya.theories.classy import classy
+from copy import deepcopy
+from typing import NamedTuple, Sequence, Union, Optional
+from cobaya.tools import load_module
+import logging
+import os
+import numpy as np
+import time
+
+class classy_sz(classy):
+    use_class_sz_fast_mode = 0 # this is passed in the yaml file
+    use_class_sz_no_cosmo_mode = 0 # this is passed in the yaml file
+    lensing_lkl = "SOLikeT"
+    # skip_background_and_thermo = True
+    # ell_factor = False # True for pyactlite and bplike, False for clik
+
+    def initialize(self):
+        """Importing CLASS from the correct path, if given, and if not, globally."""
+        self.classy_module = self.is_installed()
+        if not self.classy_module:
+            raise NotInstalledError(
+                self.log, "Could not find CLASS_SZ. Check error message above.")
+        from classy_sz import Class, CosmoSevereError, CosmoComputationError
+        global CosmoComputationError, CosmoSevereError
+        self.classy = Class()
+        super(classy,self).initialize()
+        # Add general CLASS stuff
+        self.extra_args["output"] = self.extra_args.get("output", "")
+        if "sBBN file" in self.extra_args:
+            self.extra_args["sBBN file"] = (
+                self.extra_args["sBBN file"].format(classy=self.path))
+        # Derived parameters that may not have been requested, but will be necessary later
+        self.derived_extra = []
+        self.log.info("Initialized!")
+
+        if self.use_class_sz_no_cosmo_mode == 1:
+            self.log.info("Initializing cosmology part!")
+            initial_parameters = self.extra_args.copy()
+            # print("initial_parameters:",initial_parameters)
+
+            self.classy.set(initial_parameters)
+            self.classy.compute_class_szfast()
+            self.log.info("cosmology part initialized!")
+
+
+        # print(self.lensing_lkl)
+        # exit(0)
+
+        # # class_sz default params for lkl
+        # self.extra_args["output"] = 'tSZ_1h'
+        # self.extra_args["multipoles_sz"] = 'P15'
+        # self.extra_args['nlSZ'] = 18
+
+
+    # # here modify if you want to bypass stuff in the class computation
+    # def calculate(self, state, want_derived=True, **params_values_dict):
+    #     print("Bypassing class_sz")
+
+
+
+
+    def must_provide(self, **requirements):
+
+        if "Cl_sz" in requirements:
+            # make sure cobaya still runs as it does for standard classy
+            requirements.pop("Cl_sz")
+            # specify the method to collect the new observable
+            self.collectors["Cl_sz"] = Collector(
+                    method="cl_sz", # name of the method in classy.pyx
+                    args_names=[],
+                    args=[])
+
+        if "Cl_yxg" in requirements:
+            # make sure cobaya still runs as it does for standard classy
+            requirements.pop("Cl_yxg")
+            # specify the method to collect the new observable
+            self.collectors["Cl_yxg"] = Collector(
+                    method="cl_yg", # name of the method in classy.pyx
+                    args_names=[],
+                    args=[])
+
+        if "Cl_gxg" in requirements:
+            # make sure cobaya still runs as it does for standard classy
+            requirements.pop("Cl_gxg")
+            # specify the method to collect the new observable
+            self.collectors["Cl_gxg"] = Collector(
+                    method="cl_gg", # name of the method in classy.pyx
+                    args_names=[],
+                    args=[])
+
+
+        if "Cl_gxmu" in requirements:
+            # make sure cobaya still runs as it does for standard classy
+            requirements.pop("Cl_gxmu")
+            # specify the method to collect the new observable
+            self.collectors["Cl_gxmu"] = Collector(
+                    method="cl_gm", # name of the method in classy.pyx
+                    args_names=[],
+                    args=[])
+
+
+        if "Cl_muxmu" in requirements:
+            # make sure cobaya still runs as it does for standard classy
+            requirements.pop("Cl_muxmu")
+            # specify the method to collect the new observable
+            self.collectors["Cl_muxmu"] = Collector(
+                    method="cl_mm", # name of the method in classy.pyx
+                    args_names=[],
+                    args=[])
+
+
+        if "Cl_kxg" in requirements:
+            # make sure cobaya still runs as it does for standard classy
+            requirements.pop("Cl_kxg")
+            # specify the method to collect the new observable
+            self.collectors["Cl_kxg"] = Collector(
+                    method="cl_kg", # name of the method in classy.pyx
+                    args_names=[],
+                    args=[])
+
+        if "Cl_kxmu" in requirements:
+        # make sure cobaya still runs as it does for standard classy
+            requirements.pop("Cl_kxmu")
+        # specify the method to collect the new observable
+            self.collectors["Cl_kxmu"] = Collector(
+                    method="cl_km", # name of the method in classy.pyx
+                    args_names=[],
+                    args=[])
+
+
+        if "Cl_yxmu" in requirements:
+            # make sure cobaya still runs as it does for standard classy
+            requirements.pop("Cl_yxmu")
+            # specify the method to collect the new observable
+            self.collectors["Cl_yxmu"] = Collector(
+                    method="cl_ym", # name of the method in classy.pyx
+                    args_names=[],
+                    args=[])
+        if "Cl_kgxg" in requirements:
+            # make sure cobaya still runs as it does for standard classy
+            requirements.pop("Cl_kgxg")
+            # specify the method to collect the new observable
+            self.collectors["Cl_kgxg"] = Collector(
+                    method="cl_ggamma", # name of the method in classy.pyx
+                    args_names=[],
+                    args=[])
+        if "Cl_kgxmu" in requirements:
+                # make sure cobaya still runs as it does for standard classy
+                requirements.pop("Cl_kgxmu")
+                # specify the method to collect the new observable
+                self.collectors["Cl_kgxmu"] = Collector(
+                        method="cl_kg_m", # name of the method in classy.pyx
+                        args_names=[],
+                        args=[])
+        if "Cl_IAxg" in requirements:
+            # make sure cobaya still runs as it does for standard classy
+            requirements.pop("Cl_IAxg")
+            # specify the method to collect the new observable
+            self.collectors["Cl_IAxg"] = Collector(
+                    method="cl_IA_g", # name of the method in classy.pyx
+                    args_names=[],
+                    args=[])
+
+        if "sz_binned_cluster_counts" in requirements:
+            # make sure cobaya still runs as it does for standard classy
+            requirements.pop("sz_binned_cluster_counts")
+            # specify the method to collect the new observable
+            self.collectors["sz_binned_cluster_counts"] = Collector(
+                    method="dndzdy_theoretical", # name of the method in classy.pyx
+                    args_names=[],
+                    args=[])
+
+        if "sz_unbinned_cluster_counts" in requirements:
+            # make sure cobaya still runs as it does for standard classy
+            requirements.pop("sz_unbinned_cluster_counts")
+            # specify the method to collect the new observable
+            self.collectors["sz_unbinned_cluster_counts"] = Collector(
+                    method="szcounts_ntot_rates_loglike", # name of the method in classy.pyx
+                    args_names=[],
+                    args=[])
+
+        if "cl_cib_kappa" in requirements:
+                # make sure cobaya still runs as it does for standard classy
+                requirements.pop("cl_cib_kappa")
+                # specify the method to collect the new observable
+                self.collectors["cl_cib_kappa"] = Collector(
+                        method="cl_lens_cib", # name of the method in classy.pyx
+                        args_names=[],
+                        args=[])
+        if "cl_galn_galn" in requirements:
+                # make sure cobaya still runs as it does for standard classy
+                requirements.pop("cl_galn_galn")
+                # specify the method to collect the new observable
+                self.collectors["cl_galn_galn"] = Collector(
+                        method="cl_galn_galn", # name of the method in classy.pyx
+                        args_names=[],
+                        args=[])
+        if "cl_galn_lens" in requirements:
+                # make sure cobaya still runs as it does for standard classy
+                requirements.pop("cl_galn_lens")
+                # specify the method to collect the new observable
+                self.collectors["cl_galn_lens"] = Collector(
+                        method="cl_galn_lens", # name of the method in classy.pyx
+                        args_names=[],
+                        args=[])
+
+        if "Cl_galnxtsz" in requirements:
+                # make sure cobaya still runs as it does for standard classy
+                requirements.pop("Cl_galnxtsz")
+                # specify the method to collect the new observable
+                self.collectors["Cl_galnxtsz"] = Collector(
+                        method="cl_galn_tsz", # name of the method in classy.pyx
+                        args_names=[],
+                        args=[])
+        if "Cl_galnxgallens" in requirements:
+                # make sure cobaya still runs as it does for standard classy
+                requirements.pop("Cl_galnxgallens")
+                # specify the method to collect the new observable
+                self.collectors["Cl_galnxgallens"] = Collector(
+                        method="cl_galn_gallens", # name of the method in classy.pyx
+                        args_names=[],
+                        args=[])
+        if "Cl_lensmagnxtsz" in requirements:
+                # make sure cobaya still runs as it does for standard classy
+                requirements.pop("Cl_lensmagnxtsz")
+                # specify the method to collect the new observable
+                self.collectors["Cl_lensmagnxtsz"] = Collector(
+                        method="cl_lensmagn_tsz", # name of the method in classy.pyx
+                        args_names=[],
+                        args=[])
+        if "Cl_lensmagnxgallens" in requirements:
+                # make sure cobaya still runs as it does for standard classy
+                requirements.pop("Cl_lensmagnxgallens")
+                # specify the method to collect the new observable
+                self.collectors["Cl_lensmagnxgallens"] = Collector(
+                        method="cl_lensmagn_gallens", # name of the method in classy.pyx
+                        args_names=[],
+                        args=[])
+        if "Cl_galnxIA" in requirements:
+                # make sure cobaya still runs as it does for standard classy
+                requirements.pop("Cl_galnxIA")
+                # specify the method to collect the new observable
+                self.collectors["Cl_galnxIA"] = Collector(
+                        method="cl_galn_IA", # name of the method in classy.pyx
+                        args_names=[],
+                        args=[])
+        super().must_provide(**requirements)
+
+    # get the required new observable
+    def get_Cl(self, ell_factor=False, units="FIRASmuK2"):
+        if self.use_class_sz_fast_mode:
+            return self.get_Clfast(ell_factor=ell_factor)
+        else:
+            return self._get_Cl(ell_factor=ell_factor, units=units, lensed=True)
+
+    def get_Clfast(self,ell_factor = False):
+        # print('ell_factor:',self.ell_factor)
+        # exit(0)
+        cls = {}
+        cls = deepcopy(self._current_state["Cl"])
+        # ell_factor = self.ell_factor
+        # print(cls)
+        # exit(0)
+        # print('in get clfast:',ell_factor)
+        # print(cls)
+        lcp = np.asarray(cls['ell'])
+        # print(self.lensing_lkl)
+        if ell_factor==True:
+            cls['tt'] *= (2.7255e6)**2.*(lcp*(lcp+1.))/2./np.pi
+            cls['te'] *= (2.7255e6)**2.*(lcp*(lcp+1.))/2./np.pi
+            cls['ee'] *= (2.7255e6)**2.*(lcp*(lcp+1.))/2./np.pi
+
+        if self.lensing_lkl == "ACT" and ell_factor == False:
+            cls['tt'] *= (2.7255e6)**2.
+            cls['te'] *= (2.7255e6)**2.
+            cls['ee'] *= (2.7255e6)**2.
+        # print(cls['tt'][1230])
+        # print(cls['te'][1230])
+        # print(cls['ee'][1230])
+        # exit(0)
+        if self.lensing_lkl ==  "SOLikeT":
+            cls['pp'] *= (lcp*(lcp+1.))**2./4.
+        elif self.lensing_lkl == "ACT":
+            cls['pp'] *= 1.#(lcp*(lcp+1.))**2./4.
+        else: # here for the planck lensing lkl, using lfactor option gives:
+            cls['pp'] *= (lcp*(lcp+1.))**2.*1./2./np.pi
+        return cls
+    # get the required new observable
+    def get_Cl_sz(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_sz"])
+        return cls
+
+    def get_Cl_yxg(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_yxg"])
+        return cls
+    def get_Cl_kxg(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_kxg"])
+        return cls
+    def get_Cl_gxg(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_gxg"])
+        return cls
+    def get_Cl_muxmu(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_muxmu"])
+        return cls
+    def get_Cl_gxmu(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_gxmu"])
+        return cls
+    def get_Cl_kxmu(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_kxmu"])
+        return cls
+    def get_Cl_yxmu(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_yxmu"])
+        return cls
+    def get_Cl_kgxg(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_kgxg"])
+        return cls
+    def get_Cl_kgxmu(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_kgxmu"])
+        return cls
+    def get_Cl_IAxg(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_IAxg"])
+        return cls
+    def get_cl_galn_lens(self):
+        cls = {}
+        cls = deepcopy(self._current_state["cl_galn_lens"])
+        return cls
+    def get_cl_galn_galn(self):
+        cls = {}
+        cls = deepcopy(self._current_state["cl_galn_galn"])
+        return cls
+    def get_Cl_galnxtsz(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_galnxtsz"])
+        return cls
+    def get_Cl_galnxgallens(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_galnxgallens"])
+        return cls
+    def get_Cl_lensmagnxtsz(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_lensmagnxtsz"])
+        return cls
+    def get_Cl_lensmagnxgallens(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_lensmagnxgallens"])
+        return cls
+    def get_Cl_galnxIA(self):
+        cls = {}
+        cls = deepcopy(self._current_state["Cl_galnxIA"])
+        return cls
+
+    # get the required new observable
+    def get_sz_unbinned_cluster_counts(self):
+        cls = deepcopy(self._current_state["sz_unbinned_cluster_counts"])
+        # print(cls)
+        return cls['loglike'],cls['ntot'],cls['rates']
+
+
+    # get the required new observable
+    def get_sz_binned_cluster_counts(self):
+        cls = {}
+        cls = deepcopy(self._current_state["sz_binned_cluster_counts"])
+        return cls
+
+    def get_cl_cib_kappa(self):
+        cls = {}
+        cls = deepcopy(self._current_state["cl_cib_kappa"])
+        return cls
+
+    # IMPORTANT: this method is imported from cobaya and modified to accomodate the emulators
+    def calculate(self, state, want_derived=True, **params_values_dict):
+        # Set parameters
+        params_values = params_values_dict.copy()
+        # print('\n\n')
+        # print('>>> class_sz.py: class/class_sz using params:',params_values)
+        if 'N_ncdm' in self.extra_args.keys():
+            if self.extra_args['N_ncdm'] ==  3:
+                str_mncdm = str(params_values['m_ncdm'])
+                params_values['m_ncdm'] = str_mncdm+','+str_mncdm+','+str_mncdm
+        # print('>>> class_sz.py: class/class_sz using params:',params_values)
+        # exit(0)
+        try:
+            params_values['ln10^{10}A_s'] = params_values.pop("logA")
+            self.set(params_values)
+        except KeyError:
+            self.set(params_values)
+        # Compute!
+        try:
+            if self.use_class_sz_fast_mode == 1:
+                # start = time.perf_counter()
+                if self.use_class_sz_no_cosmo_mode == 1:
+                    # print(params_values)
+                    self.classy.compute_class_sz(params_values)
+                else:
+                    self.classy.compute_class_szfast()
+                # end = time.perf_counter()
+                # print('classy_szfast took:',end-start)
+            # self.classy.compute_class_szfast()
+            # elif self.use_class_sz_no_cosmo_mode == 1:
+            #     self.classy.compute_class_sz(params_values)
+            else:
+                self.classy.compute()
+        # "Valid" failure of CLASS: parameters too extreme -> log and report
+        except self.classy_module.CosmoComputationError as e:
+            if self.stop_at_error:
+                self.log.error(
+                    "Computation error (see traceback below)! "
+                    "Parameters sent to CLASS: %r and %r.\n"
+                    "To ignore this kind of error, make 'stop_at_error: False'.",
+                    state["params"], dict(self.extra_args))
+                raise
+            else:
+                self.log.debug("Computation of cosmological products failed. "
+                               "Assigning 0 likelihood and going on. "
+                               "The output of the CLASS error was %s" % e)
+            return False
+        # CLASS not correctly initialized, or input parameters not correct
+        except self.classy_module.CosmoSevereError:
+            self.log.error("Serious error setting parameters or computing results. "
+                           "The parameters passed were %r and %r. To see the original "
+                           "CLASS' error traceback, make 'debug: True'.",
+                           state["params"], self.extra_args)
+            raise  # No LoggedError, so that CLASS traceback gets printed
+        # Gather products
+        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()
+            method = getattr(self.classy, collector.method)
+            arg_array = self.collectors[product].arg_array
+            if isinstance(arg_array, int):
+                arg_array = np.atleast_1d(arg_array)
+            if arg_array is None:
+                state[product] = method(
+                    *self.collectors[product].args, **self.collectors[product].kwargs)
+            elif isinstance(arg_array, Sequence) or isinstance(arg_array, np.ndarray):
+                arg_array = np.array(arg_array)
+                if len(arg_array.shape) == 1:
+                    # if more than one vectorised arg, assume all vectorised in parallel
+                    n_values = len(self.collectors[product].args[arg_array[0]])
+                    state[product] = np.zeros(n_values)
+                    args = deepcopy(list(self.collectors[product].args))
+                    for i in range(n_values):
+                        for arg_arr_index in arg_array:
+                            args[arg_arr_index] = \
+                                self.collectors[product].args[arg_arr_index][i]
+                        state[product][i] = method(
+                            *args, **self.collectors[product].kwargs)
+                elif len(arg_array.shape) == 2:
+                    if len(arg_array) > 2:
+                        raise NotImplementedError("Only 2 array expanded vars so far.")
+                    # Create outer combinations
+                    x_and_y = np.array(np.meshgrid(
+                        self.collectors[product].args[arg_array[0, 0]],
+                        self.collectors[product].args[arg_array[1, 0]])).T
+                    args = deepcopy(list(self.collectors[product].args))
+                    result = np.empty(shape=x_and_y.shape[:2])
+                    for i, row in enumerate(x_and_y):
+                        for j, column_element in enumerate(x_and_y[i]):
+                            args[arg_array[0, 0]] = column_element[0]
+                            args[arg_array[1, 0]] = column_element[1]
+                            result[i, j] = method(
+                                *args, **self.collectors[product].kwargs)
+                    state[product] = (
+                        self.collectors[product].args[arg_array[0, 0]],
+                        self.collectors[product].args[arg_array[1, 0]], result)
+                else:
+                    raise ValueError("arg_array not correctly formatted.")
+            elif arg_array in self.collectors[product].kwargs:
+                value = np.atleast_1d(self.collectors[product].kwargs[arg_array])
+                state[product] = np.zeros(value.shape)
+                for i, v in enumerate(value):
+                    kwargs = deepcopy(self.collectors[product].kwargs)
+                    kwargs[arg_array] = v
+                    state[product][i] = method(
+                        *self.collectors[product].args, **kwargs)
+            else:
+                raise LoggedError(self.log, "Variable over which to do an array call "
+                                            f"not known: arg_array={arg_array}")
+            if collector.post:
+                state[product] = collector.post(*state[product])
+        # Prepare derived parameters
+        d, d_extra = self._get_derived_all(derived_requested=want_derived)
+        if want_derived:
+            state["derived"] = {p: d.get(p) for p in self.output_params}
+            # Prepare necessary extra derived parameters
+        state["derived_extra"] = deepcopy(d_extra)
+        # exit(0)
+
+
+    # # get the required new observable
+    # def get_Cl(self,ell_factor=True,units="FIRASmuK2"):
+    #     ell_factor=self.ell_factor
+    #     # if self.tsz.use_class_sz_fast_mode == 1:
+    #     cls = {}
+    #     cls['ell'] = np.arange(20000)
+    #     # print(cls['ell'])
+    #     cls['tt'] = np.zeros(20000)
+    #     cls['te'] = np.zeros(20000)
+    #     cls['ee'] = np.zeros(20000)
+    #     cls['pp'] = np.zeros(20000)
+    #     # if self.tt_spectra is not None:
+    #     nl = len(self.classy.class_szfast.cp_predicted_tt_spectrum)
+    #     # print('nl:',nl)
+    #     cls['tt'][2:nl+2] = (2.7255e6)**2.*self.classy.class_szfast.cp_predicted_tt_spectrum
+    #     if ell_factor==False:
+    #         lcp = np.asarray(cls['ell'][2:nl+2])
+    #         cls['tt'][2:nl+2] *= 1./(lcp*(lcp+1.)/2./np.pi)
+    #
+    # # if self.te_spectra is not None:
+    #     cls['te'][2:nl+2] = (2.7255e6)**2.*self.classy.class_szfast.cp_predicted_te_spectrum
+    #     if ell_factor==False:
+    #         lcp = np.asarray(cls['ell'][2:nl+2])
+    #         cls['te'][2:nl+2] *= 1./(lcp*(lcp+1.)/2./np.pi)
+    # # if self.ee_spectra is not None:
+    #     cls['ee'][2:nl+2] = (2.7255e6)**2.*self.classy.class_szfast.cp_predicted_ee_spectrum
+    #     if ell_factor==False:
+    #         lcp = np.asarray(cls['ell'][2:nl+2])
+    #         cls['ee'][2:nl+2] *= 1./(lcp*(lcp+1.)/2./np.pi)
+    #     # if self.pp_spectra is not None:
+    #         # nl = len(self.pp_spectra[0])
+    #     if self.lensing_lkl ==  "SOLikeT":
+    #         cls['pp'][2:nl+2] = self.classy.class_szfast.cp_predicted_pp_spectrum/4. ## this is clkk... works for so lensinglite lkl
+    #     else:
+    #         # here for the planck lensing lkl, using lfactor option gives:
+    #         lcp = np.asarray(cls['ell'][2:nl+2])
+    #         cls['pp'][2:nl+2] = self.classy.class_szfast.cp_predicted_pp_spectrum/(lcp*(lcp+1.))**2.
+    #         cls['pp'][2:nl+2] *= (lcp*(lcp+1.))**2./2./np.pi
+    #     return cls
+
+    # def check_ranges(self, z, k):
+    #     return 1
+
+    # IMPORTANT: copied from cobaya and changed.
+    def get_param(self, p):
+        translated = self.translate_param(p)
+        for pool in ["params", "derived", "derived_extra"]:
+            value = (self.current_state[pool] or {}).get(translated, None)
+            if p == 'omegam':
+                print('getting omegam in get_param')
+                # print(translated)
+                # print(self.classy.Omega_m())
+                # exit(0)
+                return self.classy.Omega_m()
+            if value is not None:
+                return value
+
+        raise LoggedError(self.log, "Parameter not known: '%s'", p)
+### ORIGINAL FUNCTION:
+#    def get_param(self, p):
+#        translated = self.translate_param(p)
+#        for pool in ["params", "derived", "derived_extra"]:
+#            value = (self.current_state[pool] or {}).get(translated, None)
+#            if value is not None:
+#                return value
+#
+#        raise LoggedError(self.log, "Parameter not known: '%s'", p)
+
+
+
+    @classmethod
+    def is_installed(cls, **kwargs):
+        return load_module('classy_sz')
+
+
+# this just need to be there as it's used to fill-in self.collectors in must_provide:
+class Collector(NamedTuple):
+    method: str
+    args: Sequence = []
+    args_names: Sequence = []
+    kwargs: dict = {}
+    arg_array: Union[int, Sequence] = None
+    post: Optional[callable] = None