From fc7db3c989e483ff225c2b601fb87ac056acf8f1 Mon Sep 17 00:00:00 2001
From: asteri <astridklipfel@gmail.com>
Date: Wed, 22 Apr 2026 18:16:26 +0200
Subject: [PATCH] add scale node

---
 src/circuit/dag.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/circuit/dag.rs b/src/circuit/dag.rs
index 577952d..33c6a9a 100644
--- a/src/circuit/dag.rs
+++ b/src/circuit/dag.rs
@@ -11,6 +11,7 @@ new_key_type! { pub struct NodeId; }
 #[derive(Clone, Debug, PartialEq, Eq, Hash)]
 pub enum Node<V: Var> {
     Leaf(V),
+    Scale(NodeId, i32),
     Sum(NodeId, NodeId),
     Prod(NodeId, NodeId),
 }
@@ -19,6 +20,7 @@ impl<V: Var> Node<V> {
     pub fn children(&self) -> impl Iterator<Item = NodeId> {
         match self {
             Node::Leaf(_) => [None, None],
+            Node::Scale(n,_) => [Some(*n), None],
             Node::Sum(l, r) | Node::Prod(l, r) => [Some(*l), Some(*r)],
         }
         .into_iter()