ahuston-0 c4bfafeaef configure ruff, separate hydraJobs from other derivations
Signed-off-by: ahuston-0 <aliceghuston@gmail.com>
2025-03-02 01:00:04 -05:00

31 lines
833 B
Python

#!/usr/bin/env python3
from flupdt.flake_show import get_derivations
from flupdt.cli import parse_inputs
from flupdt.flake_eval import evaluate_output
import logging
import rad_development_python as rd
def main():
rd.configure_logger("INFO")
args = parse_inputs()
flake_path = args.flake_path
derivations, hydraJobs = rd.partition(
lambda s: s.startswith("hydraJobs"), get_derivations(flake_path)
)
logging.info(f"derivations: {list(derivations)}")
for d in derivations:
evaluate_output(flake_path, d)
if not args.keep_hydra:
logging.info("--keep-hydra flag is not specified, removing Hydra jobs")
else:
logging.info(f"hydraJobs: {list(hydraJobs)}")
for d in hydraJobs:
evaluate_output(flake_path, d)
if __name__ == "__main__":
main()