Skip to content

Commit b30dad2

Browse files
committed
lint
1 parent b58668c commit b30dad2

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

rdagent/scenarios/data_science/proposal/exp_gen/proposal.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,10 @@
3030
DSExperimentPlan,
3131
RD_Agent_TIMER_wrapper,
3232
)
33-
from rdagent.scenarios.data_science.proposal.exp_gen.utils import get_available_packages_prompt, get_packages
33+
from rdagent.scenarios.data_science.proposal.exp_gen.utils import (
34+
get_available_packages_prompt,
35+
get_packages,
36+
)
3437
from rdagent.utils.agent.tpl import T
3538
from rdagent.utils.repo.diff import generate_diff_from_dict
3639
from rdagent.utils.workflow import wait_retry

rdagent/scenarios/data_science/proposal/exp_gen/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -108,11 +108,11 @@ def get_packages(pkgs: list[str] | None = None) -> str:
108108
def get_available_packages_prompt() -> str:
109109
"""Generate prompt template for dynamically detected available packages."""
110110
# Use the same approach as get_packages but call the packages prompt functionality
111-
111+
112112
env = get_ds_env()
113113
implementation = FBWorkspace()
114114
fname = "package_info.py"
115115
implementation.inject_files(**{fname: (Path(__file__).absolute().resolve().parent / "package_info.py").read_text()})
116-
116+
117117
stdout = implementation.execute(env=env, entry=f"python {fname} --packages-prompt")
118118
return stdout.strip()

0 commit comments

Comments
 (0)