configure ruff, separate hydraJobs from other derivations

Signed-off-by: ahuston-0 <aliceghuston@gmail.com>
This commit is contained in:
2024-08-04 11:24:38 -04:00
parent bdc15b2c41
commit c4bfafeaef
4 changed files with 47 additions and 20 deletions

View File

@ -10,19 +10,21 @@ import rad_development_python as rd
def main():
rd.configure_logger("INFO")
args = parse_inputs()
print("hi")
flake_path = args.flake_path
derivations = get_derivations(flake_path)
if (
not args.keep_hydra
and len(list(filter(lambda s: s.startswith("hydraJobs"), derivations))) > 0
):
logging.info("--keep-hydra flag is not specified, removing Hydra jobs")
derivations = filter(lambda s: not s.startswith("hydraJobs"), derivations)
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()