fix ruff checks (mostly documentation and dead code)

This commit is contained in:
2025-03-03 17:19:12 -05:00
parent e1f672d1c5
commit 99757e4ae9
7 changed files with 119 additions and 30 deletions

39
flupdt/main.py Normal file → Executable file
View File

@ -1,10 +1,30 @@
#!/usr/bin/env python3
from flupdt.flake_show import get_derivations
from flupdt.cli import parse_inputs
from flupdt.flake_eval import evaluate_output
from flupdt.common import configure_logger, partition
"""Default processing of flake outputs for evaluating flake updates."""
import logging
from argparse import Namespace
from flupdt.cli import parse_inputs
from flupdt.common import configure_logger, partition
from flupdt.flake_build import build_output
from flupdt.flake_eval import evaluate_output
from flupdt.flake_show import get_derivations
def batch_eval(args: Namespace, flake_path: str, derivations: list[str]) -> None:
"""Bulk run evaluations or builds on a derivation set.
:params args: argument namespace to check against
:params flake_path: path to flake to be evaluated
:params derivations: list of derivations to run against
:returns None
"""
for d in derivations:
if args.evaluate:
evaluate_output(flake_path, d)
if args.build:
build_output(flake_path, d)
def main() -> None:
@ -13,23 +33,20 @@ def main() -> None:
:returns: None
"""
configure_logger(logging.DEBUG)
configure_logger("DEBUG")
args = parse_inputs()
flake_path = args.flake_path
derivations, hydra_jobs = partition(
lambda s: s.startswith("hydraJobs"), get_derivations(flake_path)
)
derivations, hydra_jobs = list(derivations), list(hydra_jobs)
logging.info(f"derivations: {list(derivations)}")
for d in derivations:
evaluate_output(flake_path, d)
batch_eval(args, flake_path, derivations)
if not args.keep_hydra:
logging.info("--keep-hydra flag is not specified, removing Hydra jobs")
else:
hydra_jobs = list(hydra_jobs)
logging.info(f"hydraJobs: {hydra_jobs}")
for d in hydra_jobs:
evaluate_output(flake_path, d)
batch_eval(args, flake_path, hydra_jobs)
if __name__ == "__main__":