From 4ba1833811a86d2584d841cd817110107f29f93f Mon Sep 17 00:00:00 2001
From: John Lancaster <32917998+jsl12@users.noreply.github.com>
Date: Fri, 20 Feb 2026 23:49:13 -0600
Subject: [PATCH] len

---
 src/daglib/dag.py | 2 +-
 src/daglib/set.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/daglib/dag.py b/src/daglib/dag.py
index 4601b72..21240bc 100644
--- a/src/daglib/dag.py
+++ b/src/daglib/dag.py
@@ -58,7 +58,7 @@ class DAG(Generic[T], MutableMapping[T, MutableSet[T]]):
         return iter(self._succ)
 
     def __len__(self) -> int:
-        return len(self._succ)
+        return sum(map(len, self._succ.values()))
 
     def __repr__(self) -> str:
         return f"{self.__class__.__name__}{dict(self._succ)!r}"
diff --git a/src/daglib/set.py b/src/daglib/set.py
index 95a3b62..fce1cbd 100644
--- a/src/daglib/set.py
+++ b/src/daglib/set.py
@@ -52,7 +52,7 @@ class DAGSet(MutableSet[T]):
     def discard(self, v: T) -> None:
         self._data.discard(v)
         if self.on_remove:
-            self.on_remove()
+            self.on_remove(v)
 
     # --- in-place operator support ---