Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions environment.yml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ channels:
dependencies:
- alabaster
- babel
- black
- black = 25.12.0
- charset-normalizer
- clmm
- compilers
Expand All @@ -29,7 +29,7 @@ dependencies:
- more-itertools
- mypy
- numcosmo >= 0.21.1
- numpy >= 2.0
- numpy >= 2.0, < 2.4
- pip >= 20.1 # pip is needed as dependency
- pip:
- cobaya
Expand Down
4 changes: 2 additions & 2 deletions examples/des_y1_3x2pt/des_y1_cosmic_shear_HMIA.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,10 +5,10 @@

import firecrown.likelihood.weak_lensing as wl
from firecrown.likelihood import TwoPoint, ConstGaussian, Likelihood
from firecrown.parameters import ParamsMap
from firecrown.updatable import ParamsMap
from firecrown.modeling_tools import ModelingTools
from firecrown.likelihood.factories import load_sacc_data
from firecrown.ccl_factory import CCLFactory
from firecrown.modeling_tools import CCLFactory
from firecrown.updatable import get_default_params_map
from firecrown.metadata_types import TracerNames

Expand Down
4 changes: 2 additions & 2 deletions examples/des_y1_3x2pt/des_y1_cosmic_shear_TATT.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,9 @@
from firecrown.likelihood.factories import load_sacc_data
import firecrown.likelihood.weak_lensing as wl
from firecrown.likelihood import TwoPoint, ConstGaussian, Likelihood
from firecrown.parameters import ParamsMap
from firecrown.updatable import ParamsMap
from firecrown.modeling_tools import ModelingTools
from firecrown.ccl_factory import CCLFactory
from firecrown.modeling_tools import CCLFactory
from firecrown.updatable import get_default_params_map
from firecrown.metadata_types import TracerNames, TRACER_NAMES_TOTAL

Expand Down
4 changes: 2 additions & 2 deletions examples/des_y1_3x2pt/des_y1_cosmic_shear_pk_modifier.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,9 @@
from firecrown.likelihood.factories import load_sacc_data
import firecrown.likelihood.weak_lensing as wl
from firecrown.likelihood import TwoPoint, ConstGaussian, Likelihood
from firecrown.parameters import ParamsMap, register_new_updatable_parameter
from firecrown.updatable import ParamsMap, register_new_updatable_parameter
from firecrown.modeling_tools import ModelingTools, PowerspectrumModifier
from firecrown.ccl_factory import CCLFactory
from firecrown.modeling_tools import CCLFactory
from firecrown.updatable import get_default_params_map
from firecrown.metadata_types import TracerNames

Expand Down
2 changes: 1 addition & 1 deletion examples/des_y1_3x2pt/factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
import firecrown.likelihood.weak_lensing as wl
import firecrown.likelihood.number_counts as nc
from firecrown.modeling_tools import ModelingTools
from firecrown.ccl_factory import CCLFactory
from firecrown.modeling_tools import CCLFactory


# The likelihood used for DES Y1 3x2pt analysis is a Gaussian likelihood, which
Expand Down
2 changes: 1 addition & 1 deletion examples/des_y1_3x2pt/factory_PM.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
import firecrown.likelihood.weak_lensing as wl
import firecrown.likelihood.number_counts as nc
from firecrown.modeling_tools import ModelingTools
from firecrown.ccl_factory import CCLFactory
from firecrown.modeling_tools import CCLFactory


# The likelihood used for DES Y1 3x2pt analysis is a Gaussian likelihood, which
Expand Down
4 changes: 2 additions & 2 deletions examples/des_y1_3x2pt/factory_PT.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,9 @@
from firecrown.likelihood.factories import load_sacc_data
from firecrown.likelihood import TwoPoint, ConstGaussian, Likelihood
from firecrown.metadata_types import TracerNames
from firecrown.parameters import ParamsMap
from firecrown.updatable import ParamsMap
from firecrown.modeling_tools import ModelingTools
from firecrown.ccl_factory import CCLFactory
from firecrown.modeling_tools import CCLFactory
from firecrown.updatable import get_default_params_map


Expand Down
2 changes: 1 addition & 1 deletion firecrown/app/analysis/_types.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
import pyccl

from firecrown.likelihood import NamedParameters
from firecrown.ccl_factory import CAMBExtraParams, PoweSpecAmplitudeParameter
from firecrown.modeling_tools import CAMBExtraParams, PoweSpecAmplitudeParameter


class Frameworks(StrEnum):
Expand Down
2 changes: 1 addition & 1 deletion firecrown/app/cosmology.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
import yaml
import pyccl

from firecrown.ccl_factory import CAMBExtraParams
from firecrown.modeling_tools import CAMBExtraParams
from . import logging
from .analysis import CCLCosmologySpec, Prior

Expand Down
2 changes: 1 addition & 1 deletion firecrown/app/examples/_cosmic_shear_template.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
from firecrown.likelihood import ConstGaussian, NamedParameters, TwoPoint
from firecrown.likelihood.factories import load_sacc_data
from firecrown.modeling_tools import ModelingTools
from firecrown.ccl_factory import CCLFactory
from firecrown.modeling_tools import CCLFactory


def build_likelihood(params: NamedParameters):
Expand Down
2 changes: 1 addition & 1 deletion firecrown/app/examples/_des_y1_3x2pt_pt_template.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
import firecrown.likelihood.number_counts as nc
from firecrown.likelihood import ConstGaussian, NamedParameters, TwoPoint
from firecrown.modeling_tools import ModelingTools
from firecrown.ccl_factory import CCLFactory
from firecrown.modeling_tools import CCLFactory


def _build_sources() -> tuple[wl.WeakLensing, nc.NumberCounts]:
Expand Down
2 changes: 1 addition & 1 deletion firecrown/app/examples/_des_y1_3x2pt_template.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
import firecrown.likelihood.number_counts as nc
from firecrown.likelihood import ConstGaussian, NamedParameters, TwoPoint
from firecrown.modeling_tools import ModelingTools
from firecrown.ccl_factory import CCLFactory
from firecrown.modeling_tools import CCLFactory


# The likelihood used for DES Y1 3x2pt analysis is a Gaussian likelihood, which
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
import firecrown.likelihood.weak_lensing as wl
from firecrown.likelihood import ConstGaussian, NamedParameters, TwoPoint
from firecrown.modeling_tools import ModelingTools
from firecrown.ccl_factory import CCLFactory
from firecrown.modeling_tools import CCLFactory


def build_likelihood(params: NamedParameters) -> tuple[ConstGaussian, ModelingTools]:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,8 @@
import firecrown.likelihood.weak_lensing as wl
from firecrown.likelihood import ConstGaussian, NamedParameters, TwoPoint
from firecrown.modeling_tools import ModelingTools, PowerspectrumModifier
from firecrown.ccl_factory import CCLFactory
from firecrown.parameters import register_new_updatable_parameter
from firecrown.modeling_tools import CCLFactory
from firecrown.updatable import register_new_updatable_parameter


# pylint: disable-next=invalid-name
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
import firecrown.likelihood.weak_lensing as wl
from firecrown.likelihood import ConstGaussian, NamedParameters, TwoPoint
from firecrown.modeling_tools import ModelingTools
from firecrown.ccl_factory import CCLFactory
from firecrown.modeling_tools import CCLFactory

# pylint: disable=duplicate-code

Expand Down
2 changes: 1 addition & 1 deletion firecrown/app/examples/_sn_srd_template.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
from firecrown.likelihood import NamedParameters, ConstGaussian
from firecrown.likelihood.factories import load_sacc_data
from firecrown.modeling_tools import ModelingTools
from firecrown.ccl_factory import CCLFactory, PoweSpecAmplitudeParameter
from firecrown.modeling_tools import CCLFactory, PoweSpecAmplitudeParameter


def build_likelihood(params: NamedParameters):
Expand Down
39 changes: 23 additions & 16 deletions firecrown/ccl_factory/__init__.py
Original file line number Diff line number Diff line change
@@ -1,37 +1,44 @@
"""This module contains the CCLFactory class and it supporting classes.
"""Deprecated: Use firecrown.modeling_tools instead.
The CCLFactory class is a factory class that creates instances of the
`pyccl.Cosmology` class.
This module is deprecated and will be removed in a future version.
All functionality has been moved to firecrown.modeling_tools.
"""

# Import all public types and classes from private submodules
from firecrown.ccl_factory._enums import (
import warnings

# Re-export everything from new location
# Import must come after warnings to emit deprecation at import time
from firecrown.modeling_tools import ( # noqa: E402
Background,
CAMBExtraParams,
CCLCalculatorArgs,
CCLCreationMode,
CCLFactory,
CCLPureModeTransferFunction,
PoweSpecAmplitudeParameter,
)
from firecrown.ccl_factory._factory import CCLFactory
from firecrown.ccl_factory._models import (
CAMBExtraParams,
CCLSplineParams,
MuSigmaModel,
PoweSpecAmplitudeParameter,
PowerSpec,
)

# Emit deprecation warning when module is imported
warnings.warn(
"firecrown.ccl_factory is deprecated and will be removed in a future version. "
"Use firecrown.modeling_tools instead.",
DeprecationWarning,
stacklevel=2,
)
from firecrown.ccl_factory._types import Background, CCLCalculatorArgs, PowerSpec

# Define __all__ for explicit API contract
# pylint: disable=duplicate-code
__all__ = [
# Type definitions
"PowerSpec",
"Background",
"CCLCalculatorArgs",
# Enum classes
"PoweSpecAmplitudeParameter",
"CCLCreationMode",
"CCLPureModeTransferFunction",
# Model classes
"MuSigmaModel",
"CAMBExtraParams",
"CCLSplineParams",
# Factory class
"CCLFactory",
]
4 changes: 2 additions & 2 deletions firecrown/connector/cobaya/likelihood.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
from cobaya.likelihood import Likelihood
from pyccl.cosmology import Pk2D

from firecrown.ccl_factory import (
from firecrown.modeling_tools import (
CCLCalculatorArgs,
CCLCreationMode,
PoweSpecAmplitudeParameter,
Expand All @@ -27,7 +27,7 @@
NamedParameters,
load_likelihood,
)
from firecrown.parameters import ParamsMap, handle_unused_params
from firecrown.updatable import ParamsMap, handle_unused_params
from firecrown.updatable import UpdatableUsageRecord
from firecrown.updatable import get_default_params_map

Expand Down
4 changes: 2 additions & 2 deletions firecrown/connector/cosmosis/likelihood.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
from cosmosis.datablock import names as section_names
from cosmosis.datablock import option_section

from firecrown.ccl_factory import CCLCreationMode
from firecrown.modeling_tools import CCLCreationMode
from firecrown.connector.mapping import MappingCosmoSIS, mapping_builder
from firecrown.likelihood import (
GaussFamily,
Expand All @@ -23,7 +23,7 @@
load_likelihood,
TwoPoint,
)
from firecrown.parameters import ParamsMap, handle_unused_params
from firecrown.updatable import ParamsMap, handle_unused_params
from firecrown.updatable import MissingSamplerParameterError, UpdatableUsageRecord


Expand Down
6 changes: 3 additions & 3 deletions firecrown/connector/mapping.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,15 +16,15 @@
from pyccl import physical_constants as physics
from typing_extensions import assert_never

from firecrown import parameters
from firecrown.ccl_factory import (
from firecrown.modeling_tools import (
Background,
CCLCalculatorArgs,
PowerSpec,
PoweSpecAmplitudeParameter,
)
from firecrown.descriptors import TypeFloat, TypeString
from firecrown.likelihood import NamedParameters
from firecrown.updatable import ParamsMap


def build_ccl_background_dict(
Expand Down Expand Up @@ -489,7 +489,7 @@ def get_params_names(
"wa",
]

def set_params_from_camb(self, params_values: parameters.ParamsMap) -> None:
def set_params_from_camb(self, params_values: ParamsMap) -> None:
"""Set the parameters in this mapping from the given CAMB-style parameters."""
# pylint: disable-msg=R0914

Expand Down
2 changes: 1 addition & 1 deletion firecrown/connector/numcosmo/helpers.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from numcosmo_py import Nc
from typing_extensions import assert_never

import firecrown.ccl_factory as fac
import firecrown.modeling_tools as fac


def get_hiprim(hi_cosmo: Nc.HICosmo) -> Nc.HIPrimPowerLaw:
Expand Down
4 changes: 2 additions & 2 deletions firecrown/connector/numcosmo/numcosmo.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
import numpy as np
from numcosmo_py import GObject, Nc, Ncm, dict_to_var_dict, var_dict_to_dict

from firecrown.ccl_factory import (
from firecrown.modeling_tools import (
CCLCalculatorArgs,
CCLCreationMode,
PoweSpecAmplitudeParameter,
Expand All @@ -23,7 +23,7 @@
load_likelihood,
)
from firecrown.modeling_tools import ModelingTools
from firecrown.parameters import ParamsMap, handle_unused_params
from firecrown.updatable import ParamsMap, handle_unused_params
from firecrown.updatable import UpdatableUsageRecord


Expand Down
16 changes: 8 additions & 8 deletions firecrown/generators/_inferred_galaxy_zdist.py
Original file line number Diff line number Diff line change
Expand Up @@ -546,10 +546,10 @@ def get_lsst_y1_lens_harmonic_bin_collection() -> ZDistLSSTSRDBinCollection:
zpu=zpu,
sigma_z=y1_lens_bins["sigma_z"],
z=RawGrid1D(values=[0.0, 3.5]),
bin_name=f"lens_{zpl:.1f}_{zpu:.1f}_y1",
bin_name=f"lsst_y1_lens{i}",
measurements={Galaxies.COUNTS},
)
for zpl, zpu in pairwise(y1_lens_bins["edges"])
for i, (zpl, zpu) in enumerate(pairwise(y1_lens_bins["edges"]))
],
)

Expand All @@ -568,10 +568,10 @@ def get_lsst_y1_source_harmonic_bin_collection() -> ZDistLSSTSRDBinCollection:
zpu=zpu,
sigma_z=y1_source_bins["sigma_z"],
z=RawGrid1D(values=[0.0, 3.5]),
bin_name=f"source_{zpl:.1f}_{zpu:.1f}_y1",
bin_name=f"lsst_y1_source{i}",
measurements={Galaxies.SHEAR_E},
)
for zpl, zpu in pairwise(y1_source_bins["edges"])
for i, (zpl, zpu) in enumerate(pairwise(y1_source_bins["edges"]))
],
)

Expand All @@ -590,10 +590,10 @@ def get_lsst_y10_lens_harmonic_bin_collection() -> ZDistLSSTSRDBinCollection:
zpu=zpu,
sigma_z=y10_lens_bins["sigma_z"],
z=RawGrid1D(values=[0.0, 3.5]),
bin_name=f"lens_{zpl:.1f}_{zpu:.1f}_y10",
bin_name=f"lsst_y10_lens{i}",
measurements={Galaxies.COUNTS},
)
for zpl, zpu in pairwise(y10_lens_bins["edges"])
for i, (zpl, zpu) in enumerate(pairwise(y10_lens_bins["edges"]))
],
)

Expand All @@ -612,10 +612,10 @@ def get_lsst_y10_source_harmonic_bin_collection() -> ZDistLSSTSRDBinCollection:
zpu=zpu,
sigma_z=y10_source_bins["sigma_z"],
z=RawGrid1D(values=[0.0, 3.5]),
bin_name=f"source_{zpl:.1f}_{zpu:.1f}_y10",
bin_name=f"lsst_y10_source{i}",
measurements={Galaxies.SHEAR_E},
)
for zpl, zpu in pairwise(y10_source_bins["edges"])
for i, (zpl, zpu) in enumerate(pairwise(y10_source_bins["edges"]))
],
)

Expand Down
Loading
Loading