dask-progress-matrix 0.0.1

Visualize Dask computations by chunk.

Install

pip install dask-progress-matrix

Quick-start

API

Use ProgressMatrix as a context manager to track Dask computations:

import dask.array as da
from dask_progress_matrix import ProgressMatrix

with ProgressMatrix(cmap="inferno"):
    da.random.random((2, 128, 256), chunks=(1, 16, 16)).compute()

CLI

Track Dask computations in any Python file using the CLI. For example, using uv:

$ uvx dask-progress-matrix compute_something.py --cmap=inferno

Features

• Terminal or Jupyter - Progress matrixes can be displayed in both terminal environments and Jupyter notebooks.

• Modes - When a computation is complete, the ProgressMatrix displays a summary with either the completed index or the elapsed time for each chunk, depending on the mode parameter.

• Data structures - Computing any Dask-backed object will display a progress matrix, including Xarray objects.

• Dimensionality - You can track the computation of any Dask array, regardless of dimensionality. For visualization, arrays are truncated to the last two dimensions, so e.g. an array with chunks (3, 16, 16) will be rendered as a 16 x 16 matrix where each chunk tracks the progress of 3 different computations.

Limitations

• Distributed schedulers - Support for distributed schedulers like dask.distributed.Client isn't currently implemented.

• Character width - Each computation chunk is rendered with a minimum width of 2 characters, so arrays with huge numbers of chunks may render slowly or poorly.

• Chunk shapes - All chunks are represented by squares, regardles of their shape.

FAQ

Why use a progress matrix?

This was mostly developed out of curiosity, but it has some practical debugging and tuning applications, like identifying chunks that are slow to compute.

Why does it take a long time to start doing anything?

The progress matrix only tracks terminal tasks that correspond directly to chunks in the computed output array. If your computation has a lot of intermediate tasks, you won't see any progress until those are completed.