ahuston-0/flake-update-diff
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove more rd, fix imports

Browse Source
This commit is contained in:
ahuston-0
2025-03-03 11:18:44 -05:00
parent 42c2f469c4
commit e1f672d1c5
4 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
flupdt/main.py
View File

@ -3,7 +3,7 @@
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
from flupdt.common import configure_logger, partition
import logging
@ -13,10 +13,10 @@ def main() -> None:
:returns: None
"""
configure_logger("INFO")
configure_logger(logging.DEBUG)
args = parse_inputs()
flake_path = args.flake_path
derivations, hydra_jobs = rd.partition(
derivations, hydra_jobs = partition(
lambda s: s.startswith("hydraJobs"), get_derivations(flake_path)
)
logging.info(f"derivations: {list(derivations)}")
@ -26,7 +26,8 @@ def main() -> None:
if not args.keep_hydra:
logging.info("--keep-hydra flag is not specified, removing Hydra jobs")
else:
logging.info(f"hydraJobs: {list(hydra_jobs)}")
hydra_jobs = list(hydra_jobs)
logging.info(f"hydraJobs: {hydra_jobs}")
for d in hydra_jobs:
evaluate_output(flake_path, d)