调整调用的最大进程数；删除wndb的封装文件

app/algorithms/burst_location/burst_location.py

@@ -21,8 +21,8 @@ from .network_model import (
     read_inf_inp_other,
 )
 
-# DEFAULT_N_WORKERS = max(1, min(cpu_count() - 1, 4))
-DEFAULT_N_WORKERS = max(1, cpu_count() - 1)
+DEFAULT_N_WORKERS = max(1, min(cpu_count() - 1, 4))
+# DEFAULT_N_WORKERS = max(1, cpu_count() - 1)
 logger = logging.getLogger(__name__)
 
 
@@ -123,7 +123,9 @@ def run_burst_location(
     partition_on_full_graph: bool = True,
     visualize_partition: bool = True,
     visualize_pause_seconds: float = 0.3,
-    final_candidates_csv_path: str | None = "temp/burst_location/final_round_candidates.csv",
+    final_candidates_csv_path: (
+        str | None
+    ) = "temp/burst_location/final_round_candidates.csv",
 ) -> dict[str, Any]:
     if pressure_scada_ids is None or len(pressure_scada_ids) == 0:
         raise ValueError("pressure_scada_ids cannot be empty.")
@@ -210,47 +212,45 @@ def run_burst_location(
             similarity_series,
             exit_condition,
             final_candidates_csv,
-        ) = (
-            DN_search_multi_simple_add_flow_count_new(
-                wn=wn,
-                wn_inp_path=str(inp_path),
-                G0=G0,
-                all_node=all_node,
-                node_x=node_x,
-                node_y=node_y,
-                pipe_start_node_all=pipe_start_node_all,
-                pipe_end_node_all=pipe_end_node_all,
-                pipe_diameter=pipe_diameter,
-                couple_node_length=couple_node_length,
-                node_pipe_dic=node_pipe_dic,
-                all_node_series=all_node_series,
-                top_group_ratio=0.3,
-                top_pipe_num_max=80,
-                top_pipe_num_min=10,
-                candidate_pipe_input_initial=candidate_pipe,
-                similarity_mode=similarity_mode,
-                pressure_monitor=pressure_monitor,
-                pressure_predict=pressure_predict,
-                pressure_normal=pressure_normal,
-                pressure_leak_all=None,
-                flow_monitor=flow_monitor,
-                flow_predict=flow_predict,
-                flow_normal=flow_normal,
-                flow_leak_all=None,
-                timestep_list=timestep_list,
-                max_flow=max_flow,
-                group_basic_num=30,
-                Top_sensor_num=min(5, len(pressure_ids)),
-                if_gy=0,
-                pressure_threshold=float(min_dpressure),
-                leak_mag=float(burst_leakage),
-                n_workers=max(1, int(n_workers)),
-                stage_timing=stage_timing,
-                partition_on_full_graph=partition_on_full_graph,
-                visualize_partition=visualize_partition,
-                visualize_pause_seconds=visualize_pause_seconds,
-                final_candidates_csv_path=final_candidates_csv_path,
-            )
+        ) = DN_search_multi_simple_add_flow_count_new(
+            wn=wn,
+            wn_inp_path=str(inp_path),
+            G0=G0,
+            all_node=all_node,
+            node_x=node_x,
+            node_y=node_y,
+            pipe_start_node_all=pipe_start_node_all,
+            pipe_end_node_all=pipe_end_node_all,
+            pipe_diameter=pipe_diameter,
+            couple_node_length=couple_node_length,
+            node_pipe_dic=node_pipe_dic,
+            all_node_series=all_node_series,
+            top_group_ratio=0.3,
+            top_pipe_num_max=80,
+            top_pipe_num_min=10,
+            candidate_pipe_input_initial=candidate_pipe,
+            similarity_mode=similarity_mode,
+            pressure_monitor=pressure_monitor,
+            pressure_predict=pressure_predict,
+            pressure_normal=pressure_normal,
+            pressure_leak_all=None,
+            flow_monitor=flow_monitor,
+            flow_predict=flow_predict,
+            flow_normal=flow_normal,
+            flow_leak_all=None,
+            timestep_list=timestep_list,
+            max_flow=max_flow,
+            group_basic_num=30,
+            Top_sensor_num=min(5, len(pressure_ids)),
+            if_gy=0,
+            pressure_threshold=float(min_dpressure),
+            leak_mag=float(burst_leakage),
+            n_workers=max(1, int(n_workers)),
+            stage_timing=stage_timing,
+            partition_on_full_graph=partition_on_full_graph,
+            visualize_partition=visualize_partition,
+            visualize_pause_seconds=visualize_pause_seconds,
+            final_candidates_csv_path=final_candidates_csv_path,
         )
     except Exception as exc:
         logger.exception("Burst location algorithm execution failed.")