//python/extensions:pip.bzl
This is the successor to pip_parse for including third party PyPI dependencies into your bazel module using bzlmod.
See also
For user documentation see the PyPI dependencies section.
- module ext pip
This extension is used to make dependencies from pip available.
pip.parse: To use, call
pip.parse()and specifyhub_nameand your requirements file. Dependencies will be downloaded and made available in a repo named after thehub_nameargument.Each
pip.parse()call configures a particular Python version. Multiple calls can be made to configure different Python versions, and will be grouped by thehub_nameargument. This allows the same logical name, e.g.@pip//numpyto automatically resolve to different, Python version-specific, libraries.pip.whl_mods: This tag class is used to help create JSON files to describe modifications to the BUILD files for wheels.
- pip.override(file, patches, patch_strip=0)
Apply any overrides (e.g. patches) to a given Python distribution defined by other tags in this extension.
- Attributes:
The Python distribution file name which needs to be patched. This will be applied to all repositories that setup this distribution via the pip.parse tag class.
mandatory
A list of patches to apply to the repository after ‘whl_library’ is extracted and BUILD.bazel file is generated.
mandatory
patch_strip– (int) (default 0)The number of leading path segments to be stripped from the file name in the patches.
optional
- pip.parse(hub_name, python_version, auth_patterns={}, download_only=False, enable_implicit_namespace_pkgs=False, environment={}, envsubst=[], experimental_extra_index_urls=[], experimental_index_url='', experimental_index_url_overrides={}, experimental_requirement_cycles={}, experimental_target_platforms=[], extra_hub_aliases={}, extra_pip_args=[], isolated=True, netrc='', parallel_download=True, pip_data_exclude=[], python_interpreter='', python_interpreter_target=None, quiet=True, requirements_by_platform={}, requirements_darwin=None, requirements_linux=None, requirements_lock=None, requirements_windows=None, timeout=600, use_hub_alias_dependencies=False, whl_modifications={})
This tag class is used to create a pip hub and all of the spokes that are part of that hub. This tag class reuses most of the attributes found in
pip_parse. The exception is it does not use the arg ‘repo_prefix’. We set the repository prefix for the user and the alias arg is always True in bzlmod.- Attributes:
The name of the repo pip dependencies will be accessible from.
This name must be unique between modules; unless your module is guaranteed to always be the root module, it’s highly recommended to include your module name in the hub name. Repo mapping,
use_repo(..., pip="my_modules_pip_deps"), can be used for shorter local names within your module.Within a module, the same
hub_namecan be specified to group different Python versions of pip dependencies under one repository name. This allows using a Python version-agnostic name when referring to pip dependencies; the correct version will be automatically selected.Typically, a module will only have a single hub of pip dependencies, but this is not required. Each hub is a separate resolution of pip dependencies. This means if different programs need different versions of some library, separate hubs can be created, and each program can use its respective hub’s targets. Targets from different hubs should not be used together.
mandatory
python_version– (str)The Python version the dependencies are targetting, in Major.Minor format (e.g., “3.11”) or patch level granularity (e.g. “3.11.1”).
If an interpreter isn’t explicitly provided (using
python_interpreterorpython_interpreter_target), then the version specified here must have a correspondingpython.toolchain()configured.mandatory
auth_patterns– (dict[str,str]) (default {})An optional dict mapping host names to custom authorization patterns.
If a URL’s host name is present in this dict the value will be used as a pattern when generating the authorization header for the http request. This enables the use of custom authorization schemes used in a lot of common cloud storage providers.
The pattern currently supports 2 tokens:
<login>and<password>, which are replaced with their equivalent value in the netrc file for the same host name. After formatting, the result is set as the value for theAuthorizationfield of the HTTP request.Example attribute and netrc for a http download to an oauth2 enabled API using a bearer token:
auth_patterns = { "storage.cloudprovider.com": "Bearer <password>" }netrc:
machine storage.cloudprovider.com password RANDOM-TOKENThe final HTTP request would have the following header:
Authorization: Bearer RANDOM-TOKEN
optional
download_only– (bool) (default False)Whether to use “pip download” instead of “pip wheel”. Disables building wheels from source, but allows use of –platform, –python-version, –implementation, and –abi in –extra_pip_args to download wheels for a different platform from the host platform.
optional
enable_implicit_namespace_pkgs– (bool) (default False)If true, disables conversion of native namespace packages into pkg-util style namespace packages. When set all py_binary and py_test targets must specify either
legacy_create_init=Falseor the global Bazel option--incompatible_default_to_explicit_init_pyto prevent__init__.pybeing automatically generated in every directory.This option is required to support some packages which cannot handle the conversion to pkg-util style.
optional
environment– (dict[str,str]) (default {})Environment variables to set in the pip subprocess. Can be used to set common variables such as
http_proxy,https_proxyandno_proxyNote that pip is run with “–isolated” on the CLI soPIP_<VAR>_<NAME>style env vars are ignored, but env vars that control requests and urllib3 can be passed. If you needPIP_<VAR>_<NAME>, take a look atextra_pip_argsandenvsubst.optional
envsubst– (list[str]) (default [])A list of environment variables to substitute (e.g.
["PIP_INDEX_URL", "PIP_RETRIES"]). The corresponding variables are expanded inextra_pip_argsusing the syntax$VARNAMEor${VARNAME}(expanding to empty string if unset) or${VARNAME:-default}(expanding to default if the variable is unset or empty in the environment). Note: On Bazel 6 and Bazel 7.0 changes to the variables named here do not cause packages to be re-fetched. Don’t fetch different things based on the value of these variables.optional
experimental_extra_index_urls– (list[str]) (default [])The extra index URLs to use for downloading wheels using bazel downloader. Each value is going to be subject to
envsubstsubstitutions if necessary.The indexes must support Simple API as described here: https://packaging.python.org/en/latest/specifications/simple-repository-api/
This is equivalent to
--extra-index-urlspipoption.optional
experimental_index_url– (str) (default “”)The index URL to use for downloading wheels using bazel downloader. This value is going to be subject to
envsubstsubstitutions if necessary.The indexes must support Simple API as described here: https://packaging.python.org/en/latest/specifications/simple-repository-api/
In the future this could be defaulted to
https://pypi.orgwhen this feature becomes stable.This is equivalent to
--index-urlpipoption.Changed in version 0.37.0: If
download_onlyis set, thensdistarchives will be discarded andpip.parsewill operate in wheel-only mode.optional
experimental_index_url_overrides– (dict[str,str]) (default {})The index URL overrides for each package to use for downloading wheels using bazel downloader. This value is going to be subject to
envsubstsubstitutions if necessary.The key is the package name (will be normalized before usage) and the value is the index URL.
This design pattern has been chosen in order to be fully deterministic about which packages come from which source. We want to avoid issues similar to what happened in https://pytorch.org/blog/compromised-nightly-dependency/.
The indexes must support Simple API as described here: https://packaging.python.org/en/latest/specifications/simple-repository-api/
optional
experimental_requirement_cycles– (dict[str,list[str]]) (default {})A mapping of dependency cycle names to a list of requirements which form that cycle.
Requirements which form cycles will be installed together and taken as dependencies together in order to ensure that the cycle is always satisified.
Example:
sphinxdepends onsphinxcontrib-serializinghtmlWhen listing both as requirements, alapy_binary( name = "doctool", ... deps = [ "@pypi//sphinx:pkg", "@pypi//sphinxcontrib_serializinghtml", ] )
Will produce a Bazel error such as
ERROR: .../external/pypi_sphinxcontrib_serializinghtml/BUILD.bazel:44:6: in alias rule @pypi_sphinxcontrib_serializinghtml//:pkg: cycle in dependency graph: //:doctool (...) @pypi//sphinxcontrib_serializinghtml:pkg (...) .-> @pypi_sphinxcontrib_serializinghtml//:pkg (...) | @pypi_sphinxcontrib_serializinghtml//:_pkg (...) | @pypi_sphinx//:pkg (...) | @pypi_sphinx//:_pkg (...) `-- @pypi_sphinxcontrib_serializinghtml//:pkg (...)Which we can resolve by configuring these two requirements to be installed together as a cycle
pip_parse( ... experimental_requirement_cycles = { "sphinx": [ "sphinx", "sphinxcontrib-serializinghtml", ] }, )
Warning: If a dependency participates in multiple cycles, all of those cycles must be collapsed down to one. For instance
a <-> banda <-> ccannot be listed as two separate cycles.optional
experimental_target_platforms– (list[str]) (default [])A list of platforms that we will generate the conditional dependency graph for cross platform wheels by parsing the wheel metadata. This will generate the correct dependencies for packages like
sphinxorpylint, which includecoloramawhen installed and used on Windows platforms.An empty list means falling back to the legacy behaviour where the host platform is the target platform.
WARNING: It may not work as expected in cases where the python interpreter implementation that is being used at runtime is different between different platforms. This has been tested for CPython only.
For specific target platforms use values of the form
<os>_<arch>where<os>is one oflinux,osx,windowsand arch is one ofx86_64,x86_32,aarch64,s390xandppc64le.You can also target a specific Python version by using
cp3<minor_version>_<os>_<arch>. If multiple python versions are specified as target platforms, then select statements of thelibandwhltargets will include usage of version aware toolchain config settings like@rules_python//python/config_settings:is_python_3.y.Special values:
host(for generating deps for the host platform only) and<prefix>_*values. For example,cp39_*,linux_*,cp39_linux_*.NOTE: this is not for cross-compiling Python wheels but rather for parsing the
whlMETADATA correctly.optional
extra_hub_aliases– (dict[str,list[str]]) (default {})Extra aliases to make for specific wheels in the hub repo. This is useful when paired with the
whl_modifications.Added in version 0.38.0.
optional
extra_pip_args– (list[str]) (default [])Extra arguments to pass on to pip. Must not contain spaces.
Supports environment variables using the syntax
$VARNAMEor${VARNAME}(expanding to empty string if unset) or${VARNAME:-default}(expanding to default if the variable is unset or empty in the environment), if"VARNAME"is listed in theenvsubstattribute. See alsoenvsubst.optional
isolated– (bool) (default True)Whether or not to pass the –isolated flag to the underlying pip command. Alternatively, the
RULES_PYTHON_PIP_ISOLATEDenvironment variable can be used to control this flag.optional
Location of the .netrc file to use for authentication
optional
parallel_download– (bool) (default True)The flag allows to make use of parallel downloading feature in bazel 7.1 and above when the bazel downloader is used. This is by default enabled as it improves the performance by a lot, but in case the queries to the simple API are very expensive or when debugging authentication issues one may want to disable this feature.
NOTE, This will download (potentially duplicate) data for multiple packages if there is more than one index available, but in general this should be negligible because the simple API calls are very cheap and the user should not notice any extra overhead.
If we are in synchronous mode, then we will use the first result that we find in case extra indexes are specified.
optional
pip_data_exclude– (list[str]) (default [])Additional data exclusion parameters to add to the pip packages BUILD file.
optional
python_interpreter– (str) (default “”)The python interpreter to use. This can either be an absolute path or the name of a binary found on the host’s
PATHenvironment variable. If no value is setpython3is defaulted for Unix systems andpython.exefor Windows.optional
python_interpreter_target– (label) (default None)If you are using a custom python interpreter built by another repository rule, use this attribute to specify its BUILD target. This allows pip_repository to invoke pip using the same interpreter as your toolchain. If set, takes precedence over python_interpreter. An example value: “@python3_x86_64-unknown-linux-gnu//:python”.
optional
If True, suppress printing stdout and stderr output to the terminal.
If you would like to get more diagnostic output, set
RULES_PYTHON_REPO_DEBUG=1orRULES_PYTHON_REPO_DEBUG_VERBOSITY=optional
requirements_by_platform– (dict[label,str]) (default {})The requirements files and the comma delimited list of target platforms as values.
The keys are the requirement files and the values are comma-separated platform identifiers. For now we only support
<os>_<cpu>values that are present in@platforms//osand@platforms//cpupackages respectively.optional
requirements_darwin– (label) (default None)Override the requirements_lock attribute when the host platform is Mac OS
optional
requirements_linux– (label) (default None)Override the requirements_lock attribute when the host platform is Linux
optional
requirements_lock– (label) (default None)A fully resolved ‘requirements.txt’ pip requirement file containing the transitive set of your dependencies. If this file is passed instead of ‘requirements’ no resolve will take place and pip_repository will create individual repositories for each of your dependencies so that wheels are fetched/built only for the targets specified by ‘build/run/test’. Note that if your lockfile is platform-dependent, you can use the
requirements_[platform]attributes.Note, that in general requirements files are compiled for a specific platform, but sometimes they can work for multiple platforms.
rules_pythonright now supports requirements files that are created for a particular platform without platform markers.optional
requirements_windows– (label) (default None)Override the requirements_lock attribute when the host platform is Windows
optional
Timeout (in seconds) on the rule’s execution duration.
optional
use_hub_alias_dependencies– (bool) (default False)Controls if the hub alias dependencies are used. If set to true, then the group_library will be included in the hub repo.
True will become default in a subsequent release.
optional
whl_modifications– (dict[label,str]) (default {})A dict of labels to wheel names that is typically generated by the whl_modifications. The labels are JSON config files describing the modifications.
optional
- pip.whl_mods(hub_name, whl_name, additive_build_content='', additive_build_content_file=None, copy_executables={}, copy_files={}, data=[], data_exclude_glob=[], srcs_exclude_glob=[])
This tag class is used to create JSON file that are used when calling wheel_builder.py. These JSON files contain instructions on how to modify a wheel’s project. Each of the attributes create different modifications based on the type of attribute. Previously to bzlmod these JSON files where referred to as annotations, and were renamed to whl_modifications in this extension.
- Attributes:
Name of the whl modification, hub we use this name to set the modifications for pip.parse. If you have different pip hubs you can use a different name, otherwise it is best practice to just use one.
You cannot have the same
hub_namein different modules. You can reuse the same name in the same module for different wheels that you put in the same hub, but you cannot have a child module that uses the samehub_name.mandatory
The whl name that the modifications are used for.
mandatory
additive_build_content– (str) (default “”)(str, optional): Raw text to add to the generated
BUILDfile of a package.optional
additive_build_content_file– (label) (default None)(label, optional): path to a BUILD file to add to the generated
BUILDfile of a package. You cannot use both additive_build_content and additive_build_content_file arguments at the same time.optional
copy_executables– (dict[str,str]) (default {})(dict, optional): A mapping of
srcandoutfiles for [@bazel_skylib//rules:copy_file.bzl][cf]. Targets generated here will also be flagged as executable.optional
copy_files– (dict[str,str]) (default {})(dict, optional): A mapping of
srcandoutfiles for [@bazel_skylib//rules:copy_file.bzl][cf]optional
data– (list[str]) (default [])(list, optional): A list of labels to add as
datadependencies to the generatedpy_librarytarget.optional
data_exclude_glob– (list[str]) (default [])(list, optional): A list of exclude glob patterns to add as
datato the generatedpy_librarytarget.optional
srcs_exclude_glob– (list[str]) (default [])(list, optional): A list of labels to add as
srcsto the generatedpy_librarytarget.optional