What s the easiest way to add a custom argument that alters Pants #general

What's the easiest way to add a custom argument th...

broad-ability-40758

07/01/2025, 12:46 PM

What's the easiest way to add a custom argument that alters the build? I want to be able to include/exclude

python_requirements

based on whether GPUs are available (via a cli argument/env var/similar) but it's not immediately obvious to me how to accomplish that

dazzling-pizza-75442

07/01/2025, 1:01 PM

maybe parametrize? https://www.pantsbuild.org/2.26/docs/using-pants/key-concepts/targets-and-build-files#parametrizing-targets

broad-ability-40758

07/01/2025, 1:03 PM

Ah yeah, that would work but be slightly inconvenient because there are a lot of 'leaf' targets (multiple docker images etc) that would all need to be altered based on the flag. Edit: But then we could use the generic target to group them which should be fine?

dazzling-pizza-75442

07/01/2025, 1:05 PM

combining with a macro or

__defaults__

may solve the problem of their are lots of places to adjust

👍 1

broad-ability-40758

07/01/2025, 1:06 PM

Cheers, will try that. IMO it would be nice if there was a lower boilerplate way to do it but it'll work.

broad-ability-40758

07/01/2025, 2:05 PM

I'd also appreciate advice on how to setup the conditional imports. I've configured a base and base + gpu resolve. In python code I need to do something like:

Copy code

if BUILT_WITH_GPU:
    import ...
else:
    shim = "blah"

I don't currently know how to tell pants the import should only be inferable if the base + gpu resolve is enabled.

9 Views

Open in Slack

Previous Next