Fix range selection when subgraph_ranges is None

TelGome · TelGome · commit 868a2160e0b8 · 2025-12-18T20:16:54.000+08:00
diff --git a/graph_net/torch/decompose_util.py b/graph_net/torch/decompose_util.py
@@ -60,7 +60,7 @@ def get_range_idx2range_by_subgraph_ranges():
 
     range_idx2range = (
         get_range_idx2range_by_split_positions()
-        if (chain_style or submodules_body_nodes is None)
+        if (subgraph_ranges is None)
         else get_range_idx2range_by_subgraph_ranges()
     )
     range_idx2submodule_body_nodes = [

Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@ def get_range_idx2range_by_subgraph_ranges():`
`60`	`60`
`61`	`61`	`range_idx2range = (`
`62`	`62`	`get_range_idx2range_by_split_positions()`
`63`		`- if (chain_style or submodules_body_nodes is None)`
	`63`	`+ if (subgraph_ranges is None)`
`64`	`64`	`else get_range_idx2range_by_subgraph_ranges()`
`65`	`65`	`)`
`66`	`66`	`range_idx2submodule_body_nodes = [`