sage.data.waveform.distributions.sky

Filename : sky.py Description : Short description of the file

Created on 2026-02-16 10:53:06

__author__ = Narenraju Nagarajan __copyright__ = Copyright 2026, ProjectName __license__ = MIT Licence __version__ = 0.0.1 __maintainer__ = Narenraju Nagarajan __affiliation__ = N/A __email__ = N/A __status__ = [‘inProgress’, ‘Archived’, ‘inUsage’, ‘Debugging’]

GitHub Repository: NULL

Documentation: NULL

Classes

UniformSky

GPU-friendly uniform sky sampler.

Module Contents

class UniformSky(polar_name='dec', azimuthal_name='ra', polar_bounds=(-torch.pi / 2, torch.pi / 2), azimuthal_bounds=(0.0, 2 * torch.pi))

GPU-friendly uniform sky sampler.

• Polar angle = declination dec in [-pi/2, pi/2]

• Azimuthal angle = right ascension ra in [0, 2pi)

polar_name = 'dec'

azimuthal_name = 'ra'

polar_sampler

azimuth_sampler

sample(shape, device=None, dtype=torch.float32, generator=None)

Draw shape sky-position pairs and return them as a dict.

Returns:

{polar_name: dec, azimuthal_name: ra} each of shape shape.

Return type:

dict[str, torch.Tensor]