diff --git a/src/core/pipeline/runner.py b/src/core/pipeline/runner.py index b4f7daf..5cf54f5 100644 --- a/src/core/pipeline/runner.py +++ b/src/core/pipeline/runner.py @@ -116,10 +116,10 @@ PIPELINE_STEPS: List[StepSpec] = [ ), StepSpec( step_id="step8", method_name="step8_water_quality_indices", - requires=["training_csv_path"], produces=["indices_path"], + requires=["training_csv_path"], produces=["indices_path", "trad_indices_dir"], required_input_files=["training_csv_path"], - output_file="{work_dir}/6_water_quality_indices/water_quality_indices.csv", - description="水质光谱指数计算(optional)", + output_file="{work_dir}/6_water_quality_indices/training_spectra_indices.csv", + description="水质光谱指数计算(双轨输出:A轨宽表 + B轨单文件)", ), StepSpec( step_id="step7", method_name="step7_ml_modeling", diff --git a/src/core/steps/data_preparation_step.py b/src/core/steps/data_preparation_step.py index ae8dcbc..3e5e1ea 100644 --- a/src/core/steps/data_preparation_step.py +++ b/src/core/steps/data_preparation_step.py @@ -161,7 +161,7 @@ class DataPreparationStep: if output_file: output_path = str(Path(output_file)) else: - output_path = str(output_dir / "water_quality_indices.csv") + output_path = str(output_dir / "training_spectra_indices.csv") if Path(output_path).exists(): print(f"检测到已存在的水质指数文件,直接使用: {output_path}")