From 76d697ce3f1e717d21364acc25274216f3d59c3e Mon Sep 17 00:00:00 2001 From: DXC Date: Mon, 22 Jun 2026 17:15:59 +0800 Subject: [PATCH] =?UTF-8?q?fix(viz=5Fthread):=20generate=5Fall=5Fselected?= =?UTF-8?q?=20=E8=AF=BB=20self.extra=20=E6=B3=A8=E5=85=A5=E8=B7=AF?= =?UTF-8?q?=E5=BE=84=EF=BC=88v1=20GUI=20=E5=AE=9E=E9=99=85=E5=89=AF?= =?UTF-8?q?=E6=9C=AC=EF=BC=89?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/gui/core/viz_thread.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/src/gui/core/viz_thread.py b/src/gui/core/viz_thread.py index ca657e9..fc1ac06 100644 --- a/src/gui/core/viz_thread.py +++ b/src/gui/core/viz_thread.py @@ -209,11 +209,19 @@ class VisualizationWorkerThread(QThread): viz = WaterQualityVisualization(output_dir=str(wp / "14_visualization")) 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 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()): from src.core.visualization.scatter_plot import generate_model_scatter_plots scatter_paths = generate_model_scatter_plots(