diff --git a/rdagent/components/coder/data_science/raw_data_loader/prompts.yaml b/rdagent/components/coder/data_science/raw_data_loader/prompts.yaml index a0976c1f..be607191 100644 --- a/rdagent/components/coder/data_science/raw_data_loader/prompts.yaml +++ b/rdagent/components/coder/data_science/raw_data_loader/prompts.yaml @@ -68,6 +68,7 @@ spec: 4. Notes: - Update `DT` (data type) based on the specific competition dataset. This can include `pd.DataFrame`, `np.array`, `torch.Tensor`, etc. - Extend domain-specific handling steps based on the competition information. + - Never use sample submission as the test index, as it may not be the same as the test data. Use the test index file or test data source to get the test index. {% if latest_spec %} 5. Former Specification: diff --git a/rdagent/scenarios/data_science/dev/runner.py b/rdagent/scenarios/data_science/dev/runner.py index 2f34c4fb..b9fc96cf 100644 --- a/rdagent/scenarios/data_science/dev/runner.py +++ b/rdagent/scenarios/data_science/dev/runner.py @@ -50,7 +50,7 @@ class DSRunner(Developer[DSExperiment]): if unused_files: logger.warning(f"Unused scripts: {unused_files}") exp.experiment_workspace.inject_files( - {file_path.name: exp.experiment_workspace.DEL_KEY for file_path in unused_files} + **{file_path.name: exp.experiment_workspace.DEL_KEY for file_path in unused_files} ) os.remove(exp.experiment_workspace.workspace_path / "coverage.json") return exp