paltas

Strong lens substructure package.


Keywords
astrophysics, cosmology, hierarchical-model, neural-network, simulation-based-inference
License
MIT
Install
pip install paltas==0.0.3

Documentation

logo paltas

pypi ci coverage Documentation Status arxiv license

paltas is a package for conducting simulation-based inference on strong gravitational lensing images. The package builds on lenstronomy to create large datasets of strong lensing images with realistic low-mass halos, Hubble Space Telescope (HST) observational effects, and galaxy light from HST's COSMOS field. paltas also includes the capability to easily train neural posterior estimators of the parameters of the lensing system and to run hierarchical inference on test populations.

Installation

paltas is installable via pip:

$ pip install paltas

The default paltas requirements do not include tensorflow, but if you are interested in using the modules contained in the Analysis folder, you will have to install tensorflow:

$ pip install tensorflow

Usage

The main functionality of paltas is to generate realistic datasets of strong gravitational lenses in a way that's modular, scalable, and user-friendly. To make a dataset with paltas all you need is a configuration file which you can then pass to the generate.py script:

$ python generate.py path/to/config/file path/to/output/folder --n 100

Running the line of code above would generate 100 lenses and output them in the specified folder. paltas comes preloaded with a number of configuration files which are described in Configs/Examples/README.rst. For example, to create a dataset with HST observational effects, subhalos, and line-of-sight halos run:

$ python generate.py Configs/Examples/config_all.py example --n 100

We provide a tutorial notebook that describes how to generate your own config file.

Demos

paltas comes with a tutorial notebook for users interested in modifying the simulation classes.

Figures

Code for generating the plots included in some of the publications using paltas can be found under the corresponding arxiv number in the notebooks/papers/ folder.

Contributing

We welcome improvements via GitHub pull request! The workflow for contributing to paltas is the same as it is for lenstronomy, please see the lenstronomy CONTRIBUTING guidelines. for the rundown.

In brief, to contribute to paltas you will need to fork the repo on GitHub, clone it, and pip install with editable option, like this:

$ git clone git@github.com:<your-account>/paltas.git
$ cd paltas
$ pip install -e .

You should then work in a local development branch, while keeping your main branch up to date with the base repo on GitHub. (That means you'll need to git remote add base https://github.com/swagnercarena/paltas.git so that you can git pull base main from your main branch).

See the lenstronomy CONTRIBUTING guidelines. for help with python style, docstring formats, creating pull requests, and so on. We recommend that you start your pull request (PR) when you _begin work on a new feature or bug fix, so that the rest of the paltas community can see what you are doing and coordinate with you - when your improvement is ready you can request code review from others via the PR interface. Try and restrict PRs to one feature per branch, to help keep them small and manageable. (Note that paltas uses Google docstring style.)

Attribution

If you use paltas or its datasets for your own research, please cite the paltas package (Wagner-Carena et al. 2022) as well as the lenstronomy package (Birrer & Amara 2018, Birrer et al. 2021).

Zenodo Uploads

The following is a list of the zenodo uploads associated to papers using paltas. These uploads will include additional chains, test sets, and model weights required to reproduce the paper results.

image