Don't delay computations on +: and :+

jcazevedo · Nov 28, 2022 · 9417117 · 9417117
1 parent 65b8ab3
commit 9417117
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/scala/net/jcazevedo/phalanges/FingerTree.scala b/src/main/scala/net/jcazevedo/phalanges/FingerTree.scala
@@ -41,7 +41,7 @@ sealed abstract class FingerTree[V, A](implicit measured: Measured[A, V]) {
           one = (_, b) => FingerTree.deep(Digit(a, b), m, sf),
           two = (_, b, c) => FingerTree.deep(Digit(a, b, c), m, sf),
           three = (_, b, c, d) => FingerTree.deep(Digit(a, b, c, d), m, sf),
-          four = (_, b, c, d, e) => FingerTree.deep(Digit(a, b), m.flatMap(ft => Lazy.delay(Node(c, d, e) +: ft)), sf)
+          four = (_, b, c, d, e) => FingerTree.deep(Digit(a, b), m.map(ft => Node(c, d, e) +: ft), sf)
         )
     )
 
@@ -54,7 +54,7 @@ sealed abstract class FingerTree[V, A](implicit measured: Measured[A, V]) {
           one = (_, b) => FingerTree.deep(pr, m, Digit(b, a)),
           two = (_, c, b) => FingerTree.deep(pr, m, Digit(c, b, a)),
           three = (_, d, c, b) => FingerTree.deep(pr, m, Digit(d, c, b, a)),
-          four = (_, e, d, c, b) => FingerTree.deep(pr, m.flatMap(ft => Lazy.delay(ft :+ Node(e, d, c))), Digit(b, a))
+          four = (_, e, d, c, b) => FingerTree.deep(pr, m.map(ft => ft :+ Node(e, d, c)), Digit(b, a))
         )
     )