Beaker-py

A lightweight pure-Python client for Beaker.

Quick Links

📝 Docs and examples:
🌍 Ecosystem:
- beaker-gantry
- beaker-run-action

Installing

Installing with `pip`

beaker-py is available on PyPI. Just run

pip install beaker-py

Installing from source

To install beaker-py from source, first clone the repository:

git clone https://github.com/allenai/beaker.git

Then create or activate a Python virtual environment, and run:

cd beaker/bindings/python
make dev-install

Quick start

If you've already configured the Beaker command-line client, beaker-py will find and use the existing configuration file (usually located at $HOME/.beaker/config.yml) or BEAKER_TOKEN environment variable.

Then you can instantiate the Beaker client with the .from_env() class method:

from beaker import Beaker

with Beaker.from_env() as beaker:
    ...

With the Python client, you can:

Query Clusters with beaker.cluster.* methods. For example:
```
beaker.cluster.get("ai2/jupiter-cirrascale-2")
```
Manage Datasets with beaker.dataset.* methods. For example:
```
beaker.dataset.create(dataset_name, source_dir)
```
Manage Experiments with beaker.experiment.* and beaker.workload.* methods. For example:
```
beaker.experiment.create(spec=spec, name=name)
```
Manage Groups with beaker.group.* methods. For example:
```
beaker.group.create(name)
```
Manage Images with beaker.image.* methods. For example:
```
beaker.image.update(image, name=name)
```
Manage Secrets with beaker.secret.* methods. For example:
```
beaker.secret.write(name, value)
```
Manage Workspaces with beaker.workspace.* methods. For example:
```
beaker.workspace.create("ai2/new_workspace")
```
Track Jobs with beaker.job.* methods. For example:
```
beaker.job.logs(job, follow=True)
```
Create and process Queues with beaker.queue.* methods. For example:
```
beaker.queue.create("my-work-queue", batch_size=4)
```

If you're coming from v1 of beaker-py, consider reading the migration guide.

Example workflow

Launch and follow an experiment like beaker-gantry does:

import time
from beaker import Beaker, BeakerExperimentSpec, BeakerJobPriority


with Beaker.from_env() as beaker:
    # Build experiment spec...
    spec = BeakerExperimentSpec.new(
        description="beaker-py test run",
        beaker_image="petew/hello-world",
        priority=BeakerJobPriority.low,
        min_runtime="1h",
    )

    # Create experiment workload...
    workload = beaker.experiment.create(spec=spec)

    # Wait for job to be created...
    while (job := beaker.workload.get_latest_job(workload)) is None:
        print("waiting for job to start...")
        time.sleep(1.0)

    # Follow logs...
    print("Job logs:")
    for job_log in beaker.job.logs(job, follow=True):
        print(job_log.message.decode())

See the integration tests for more examples.

Development

After installing from source, set BEAKER_TOKEN in your environment.

Then, you can run checks and tests locally with:

make checks

Releases

At the moment releases need to be published manually by following these steps:

Ensure you've authenticated with PyPI through a ~/.pypirc file and have write permissions to the beaker-py project. The local ~/.pypirc file looks like:

[distutils]
index-servers = pypi

[pypi]
username = __token__
password = #beaker-py pypi API token from 1pass#

Ensure the target release version defined in src/beaker/version.py is correct, or change the version on the fly by adding the Make argument BEAKER_PY_VERSION=X.X.X to the command in the next step.
Ensure the CHANGELOG.md has a section at the top for the new release (## vX.X.X - %Y-%m-%d).
Run make publish for a stable release or make publish-nightly for a nightly pre-release.

Release files for beaker-py 2.8.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for beaker-py 2.8.0
File	Size	Uploaded
beaker_py-2.8.0.tar.gz	113.2 kB	Sep 9, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for beaker-py 2.8.0
File	Interpreter	ABI	Platform	Reset
beaker_py-2.8.0-py3-none-any.whl 124.5 kB Sep 9, 2026	Python 3	none	any	Details

Total release size: 237.7 kB

Release files / beaker_py-2.8.0.tar.gz

Download URL	beaker_py-2.8.0.tar.gz
Size	113.2 kB
Tags	Source
SHA-256 checksum How to use checksums	`aae9421307ae37753ac371fd2be162db148e489e2932cc54a695168e27a208e1`
BLAKE2b-256 checksum How to use checksums	`638503804652e4af9fcfbe2ff80929b98d9be3027a5799b94deeb2d027663fba`
Upload date	Sep 9, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/6.2.0 CPython/3.11.9`

Release files / beaker_py-2.8.0-py3-none-any.whl

Download URL	beaker_py-2.8.0-py3-none-any.whl
Size	124.5 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`e69c055919755a43da8c00a5eb813249293d4195c76ff3ee614e10095f455f58`
BLAKE2b-256 checksum How to use checksums	`526e7ee35bcfabe8853e5bd12766655c5b37ff87d509598926de16cccc863517`
Upload date	Sep 9, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/6.2.0 CPython/3.11.9`