diff --git a/uv/pip.bzl b/uv/pip.bzl index eaec087..554f6bb 100644 --- a/uv/pip.bzl +++ b/uv/pip.bzl @@ -13,6 +13,7 @@ def pip_compile( args = None, data = None, tags = None, + size = None, **kwargs): """ Produce targets to compile a requirements.in or pyproject.toml file into a requirements.txt file. @@ -41,6 +42,7 @@ def pip_compile( requirements_in = requirements_in or "//:requirements.in" requirements_txt = requirements_txt or "//:requirements.txt" tags = tags or [] + size = size or "small" if types.is_list(requirements_in): write_target = "_{}.write".format(name) write_file( @@ -78,5 +80,6 @@ def pip_compile( data = data, uv_args = args, tags = ["requires-network"] + tags, + size = size, **kwargs )