graph/: import Node & friends from defining modules, not the package

The safe-autofix pass alphabetized graph/__init__.py, which exposed that
sibling modules imported Node/Concatenate/LiteralValue from the package
__init__ and relied on its hand-tuned import order. Import from the
defining modules (graph.node, graph.misc) instead - order-independent
and cycle-free.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>

Author: physicsrob

torchwright/graph/asserts.py

@@ -43,9 +43,8 @@
 
 import torch
 
-from torchwright.graph import Concatenate, Node
-from torchwright.graph.misc import Check, Predicate
-from torchwright.graph.node import _current_annotation
+from torchwright.graph.misc import Check, Concatenate, Predicate
+from torchwright.graph.node import Node, _current_annotation
 from torchwright.graph.value_type import NodeValueType, Range, tightened_with
 
 

torchwright/graph/attn.py

@@ -1,6 +1,6 @@
 import torch
 
-from torchwright.graph import Node
+from torchwright.graph.node import Node
 from torchwright.graph.rope import ROPE_BASE, apply_rope, rope_cos_sin
 
 # Causal mask sentinel: future positions are filled with this value before

torchwright/graph/embedding.py

@@ -1,6 +1,6 @@
 import torch
 
-from torchwright.graph import Node
+from torchwright.graph.node import Node
 from torchwright.graph.spherical_codes import get_spherical_codes
 from torchwright.graph.value_type import NodeValueType
 

torchwright/graph/ffn.py

@@ -24,7 +24,7 @@
 
 import torch
 
-from torchwright.graph import Node
+from torchwright.graph.node import Node
 from torchwright.graph.value_type import NodeValueType
 
 

torchwright/graph/linear.py

@@ -1,6 +1,6 @@
 import torch
 
-from torchwright.graph import Node
+from torchwright.graph.node import Node
 from torchwright.graph.value_type import NodeValueType
 
 

torchwright/graph/misc.py

@@ -3,7 +3,7 @@
 
 import torch
 
-from torchwright.graph import Node
+from torchwright.graph.node import Node
 from torchwright.graph.value_type import NodeValueType
 
 # A predicate maps a value tensor to ``(ok, detail)`` where ``detail`` is a

torchwright/graph/optimize.py

@@ -9,9 +9,10 @@
 
 import torch
 
-from torchwright.graph import Concatenate, LiteralValue, Node
 from torchwright.graph.ffn import FFN
 from torchwright.graph.linear import Linear
+from torchwright.graph.misc import Concatenate, LiteralValue
+from torchwright.graph.node import Node
 
 
 @dataclass

torchwright/graph/relu.py

@@ -1,6 +1,6 @@
 import torch
 
-from torchwright.graph import Node
+from torchwright.graph.node import Node
 from torchwright.graph.value_type import NodeValueType