fix ruff checks (mostly documentation and dead code)
This commit is contained in:
39
flupdt/main.py
Normal file → Executable file
39
flupdt/main.py
Normal file → Executable 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__":
|
||||
|
Reference in New Issue
Block a user