At the next Pants major release version, BUILD files won't support arbitrary python code, they'll be purely declarative. So avoiding code if at all possible will help you in the long run.