rail.plotting.pz_data_holders module

class rail.plotting.pz_data_holders.RailPZMultiPointEstimateDataHolder(**kwargs)

Bases: RailDatasetHolder

Simple class for holding making a merged for plotting data that comes from a RailProject

Parameters:

kwargs (Any)

config_options: dict[str, StageParameter] = {'datasets': Parameter(Dataset name, type: <class 'list'>, default: None [required]), 'name': Parameter(Dataset name, type: <class 'str'>, default: None [required])}

extractor_inputs: dict = {'datasets': list[rail.plotting.dataset_holder.RailDatasetHolder]}

get_extractor_inputs()

Resolve the inputs needed to get the data from the configuration paramters.

For example, load RailProject configurations, resolve the set of requested interpolants, etc…

Return type:

dict[str, Any]

output_type

alias of RailPZMultiPointEstimateDataset

class rail.plotting.pz_data_holders.RailPZPointEstimateDataHolder(**kwargs)

Bases: RailDatasetHolder

Class to extract true redshifts and one p(z) point estimate from a RailProject.

This will return a dict:

truth: np.ndarray

True redshifts

pointEstimate: np.ndarray

Point estimates of the true redshifts

Parameters:

kwargs (Any)

config_options: dict[str, StageParameter] = {'algo': Parameter(RailProject algorithm, type: <class 'str'>, default: None [required]), 'flavor': Parameter(RailProject analysis flavor, type: <class 'str'>, default: None [required]), 'name': Parameter(Dataset name, type: <class 'str'>, default: None [required]), 'project': Parameter(RailProject name, type: <class 'str'>, default: None [required]), 'selection': Parameter(RailProject data selection, type: <class 'str'>, default: None [required]), 'tag': Parameter(RailProject file tag, type: <class 'str'>, default: None [required])}

extractor_inputs: dict = {'algo': <class 'str'>, 'flavor': <class 'str'>, 'project': <class 'rail.projects.project.RailProject'>, 'selection': <class 'str'>, 'tag': <class 'str'>}

classmethod generate_dataset_dict(**kwargs)

Return type:

tuple[list[RailProjectHolder], list[RailDatasetHolder], list[RailDatasetListHolder]]

Parameters:

**kwargs (Any) – Set Notes

Notes

dataset_list_name: str

Name for the resulting DatasetList

project_file: str

Config file for project to inspect

selections: list[str]

Selections to use

flavors: list[str]

Flavors to use

Returns:

• list[RailProjectHolder] – Underlying RailProjects

• list[RailDatasetHolder] – Extracted datasets

• list[RailDatasetListHolder] – Extracted dataset lists

Parameters:

kwargs (Any)

Return type:

tuple[list[RailProjectHolder], list[RailDatasetHolder], list[RailDatasetListHolder]]

get_extractor_inputs()

Resolve the inputs needed to get the data from the configuration paramters.

For example, load RailProject configurations, resolve the set of requested interpolants, etc…

Return type:

dict[str, Any]

output_type

alias of RailPZPointEstimateDataset