fix(viz_thread): generate_all_selected 读 self.extra 注入路径(v1 GUI 实际副本)

This commit is contained in:
DXC
2026-06-22 17:15:59 +08:00
parent 56d8a292be
commit 76d697ce3f

View File

@ -209,11 +209,19 @@ class VisualizationWorkerThread(QThread):
viz = WaterQualityVisualization(output_dir=str(wp / "14_visualization")) viz = WaterQualityVisualization(output_dir=str(wp / "14_visualization"))
parts = [] parts = []
training_csv = wp / "5_training_spectra" / "training_spectra.csv" training_csv_path = (self.extra.get("training_csv_path") or "").strip()
if training_csv_path:
training_csv = Path(training_csv_path)
else:
training_csv = wp / "5_training_spectra" / "training_spectra.csv"
if self.extra.get("gen_scatter"): if self.extra.get("gen_scatter"):
if training_csv.is_file(): if training_csv.is_file():
models_dir = wp / "7_Supervised_Model_Training" models_dir_str = (self.extra.get("models_dir") or "").strip()
if models_dir_str:
models_dir = Path(models_dir_str)
else:
models_dir = wp / "7_Supervised_Model_Training"
if models_dir.is_dir() and any(d.is_dir() for d in models_dir.iterdir()): if models_dir.is_dir() and any(d.is_dir() for d in models_dir.iterdir()):
from src.core.visualization.scatter_plot import generate_model_scatter_plots from src.core.visualization.scatter_plot import generate_model_scatter_plots
scatter_paths = generate_model_scatter_plots( scatter_paths = generate_model_scatter_plots(