tensorflow/third_party/py/python_init_repositories.bzl
Vadym Matsishevskyi 5dd1758095 Add support for local wheel whitelisting and blacklisting
Also fix python version matching logic for wheels which do not require a specific python version.

PiperOrigin-RevId: 650383841
2024-07-08 15:23:41 -07:00

24 lines
964 B
Python

"""Hermetic Python initialization. Consult the WORKSPACE on how to use it."""
load("@rules_python//python:repositories.bzl", "py_repositories")
load("//third_party/py:python_repo.bzl", "python_repository")
def python_init_repositories(
requirements = {},
local_wheel_workspaces = [],
local_wheel_dist_folder = None,
default_python_version = None,
local_wheel_inclusion_list = ["*"],
local_wheel_exclusion_list = []):
python_repository(
name = "python_version_repo",
requirements_versions = requirements.keys(),
requirements_locks = requirements.values(),
local_wheel_workspaces = local_wheel_workspaces,
local_wheel_dist_folder = local_wheel_dist_folder,
default_python_version = default_python_version,
local_wheel_inclusion_list = local_wheel_inclusion_list,
local_wheel_exclusion_list = local_wheel_exclusion_list,
)
py_repositories()