From e1f672d1c5bb4c3695c7469b4ee5b876f6cd8791 Mon Sep 17 00:00:00 2001
From: ahuston-0 <aliceghuston@gmail.com>
Date: Mon, 3 Mar 2025 11:18:44 -0500
Subject: [PATCH] remove more rd, fix imports

---
 flupdt/common.py     | 1 +
 flupdt/flake_eval.py | 2 ++
 flupdt/flake_show.py | 1 +
 flupdt/main.py       | 9 +++++----
 4 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/flupdt/common.py b/flupdt/common.py
index 5e2abe2..f9b9d65 100644
--- a/flupdt/common.py
+++ b/flupdt/common.py
@@ -3,6 +3,7 @@
 import itertools
 import logging
 import sys
+from subprocess import Popen,PIPE
 
 
 def configure_logger(level: str = "INFO") -> None:
diff --git a/flupdt/flake_eval.py b/flupdt/flake_eval.py
index 316caa3..d143e86 100644
--- a/flupdt/flake_eval.py
+++ b/flupdt/flake_eval.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 from typing import Optional
 from flupdt.common import bash_wrapper
diff --git a/flupdt/flake_show.py b/flupdt/flake_show.py
index 2fbddbf..c9a56be 100644
--- a/flupdt/flake_show.py
+++ b/flupdt/flake_show.py
@@ -5,6 +5,7 @@ import logging
 import re
 import shutil
 import typing
+from subprocess import Popen
 
 from flupdt.common import bash_wrapper
 
diff --git a/flupdt/main.py b/flupdt/main.py
index fe3f2e2..497c357 100644
--- a/flupdt/main.py
+++ b/flupdt/main.py
@@ -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)
 
-- 
2.48.1