remove more rd, fix imports #2

Merged
ahuston-0 merged 1 commits from feature/rd into main 2025-03-03 11:19:22 -05:00
4 changed files with 9 additions and 4 deletions

View File

@ -3,6 +3,7 @@
import itertools
import logging
import sys
from subprocess import Popen,PIPE
def configure_logger(level: str = "INFO") -> None:

View File

@ -1,3 +1,5 @@
from __future__ import annotations
import logging
from typing import Optional
from flupdt.common import bash_wrapper

View File

@ -5,6 +5,7 @@ import logging
import re
import shutil
import typing
from subprocess import Popen
from flupdt.common import bash_wrapper

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)