Skip to content

A high performance Python graph library implemented in Rust.

License

Notifications You must be signed in to change notification settings

kevinhartman/rustworkx

 
 

Repository files navigation

retworkx

License Build Status Build Status Coverage Status Minimum rustc 1.41.1 arXiv

retworkx is a general purpose graph library for python3 written in Rust to take advantage of the performance and safety that Rust provides. It was built as a replacement for qiskit's previous (and current) networkx usage (hence the name) but is designed to provide a high performance general purpose graph library for any python application. The project was originally started to build a faster directed graph to use as the underlying data structure for the DAG at the center of qiskit-terra's transpiler, but it has since grown to cover all the graph usage in Qiskit and other applications.

Installing retworkx

retworkx is published on pypi so on x86_64, i686, ppc64le, s390x, and aarch64 Linux systems, x86_64 on Mac OSX, and 32 and 64 bit Windows installing is as simple as running:

pip install retworkx

This will install a precompiled version of retworkx into your python environment.

Installing on a platform without precompiled binaries

If there are no precompiled binaries published for your system you'll have to build the package from source. However, to be able able to build the package from the published source package you need to have rust >=1.41.1 installed (and also cargo which is normally included with rust) You can use rustup (a cross platform installer for rust) to make this simpler, or rely on other installation methods. A source package is also published on pypi, so you still can also run the above pip command to install it. Once you have rust properly installed, running:

pip install retworkx

will build retworkx for your local system from the source package and install it just as it would if there was a prebuilt binary available.

Optional dependencies

If you're planning to use the retworkx.visualization module you will need to install optional dependencies to use the functions. The matplotlib based drawer function retworkx.visualization.mpl_draw requires that the matplotlib library is installed. This can be installed with pip install matplotlib or when you're installing retworkx with pip install 'retworkx[mpl]'. If you're going to use the graphviz based drawer function retworkx.visualization.graphviz_drawer first you will need to install graphviz, instructions for this can be found here: https://graphviz.org/download/#executable-packages. Then you will need to install the pydot and pillow Python libraries. This can be done either with pip install pydot pillow or when installing retworkx with pip install 'retworkx[graphviz]'.

If you would like to install all the optional Python dependencies when you install retworkx you can use pip install 'retworkx[all]' to do this.

Using retworkx

Once you have retworkx installed you can use it by importing retworkx. All the functions and graph classes are off the root of the package. For example, calculating the shortest path between A and C would be:

import retworkx

graph = retworkx.PyGraph()

# Each time add node is called, it returns a new node index
a = graph.add_node("A")
b = graph.add_node("B")
c = graph.add_node("C")

# add_edges_from takes tuples of node indices and weights,
# and returns edge indices
graph.add_edges_from([(a, b, 1.5), (a, c, 5.0), (b, c, 2.5)])

# Returns the path A -> B -> C
retworkx.dijkstra_shortest_paths(graph, a, c, weight_fn=float)

Building from source

The first step for building retworkx from source is to clone it locally with:

git clone https://github.com/Qiskit/retworkx.git

retworkx uses PyO3 and setuptools-rust to build the python interface, which enables using standard python tooling to work. So, assuming you have rust installed, you can easily install retworkx into your python environment using pip. Once you have a local clone of the repo, change your current working directory to the root of the repo. Then, you can install retworkx into your python env with:

pip install .

Assuming your current working directory is still the root of the repo. Otherwise you can run:

pip install $PATH_TO_REPO_ROOT

which will install it the same way. Then retworkx is installed in your local python environment. There are 2 things to note when doing this though, first if you try to run python from the repo root using this method it will not work as you expect. There is a name conflict in the repo root because of the local python package shim used in building the package. Simply run your python scripts or programs using retworkx outside of the repo root. The second issue is that any local changes you make to the rust code will not be reflected live in your python environment, you'll need to recompile retworkx by rerunning pip install to have any changes reflected in your python environment.

Develop Mode

If you'd like to build retworkx in debug mode and use an interactive debugger while working on a change you can use python setup.py develop to build and install retworkx in develop mode. This will build retworkx without optimizations and include debuginfo which can be handy for debugging. Do note that installing retworkx this way will be significantly slower then using pip install and should only be used for debugging/development.

It's worth noting that pip install -e does not work, as it will link the python packaging shim to your python environment but not build the retworkx binary. If you want to build retworkx in debug mode you have to use python setup.py develop.

Authors and Citation

retworkx is the work of many people who contribute to the project at different levels. If you use retworkx in your research, please cite our paper as per the included BibTeX file.

About

A high performance Python graph library implemented in Rust.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 52.0%
  • Rust 47.8%
  • Shell 0.2%