fix(transducers): flattenWith()

thi-ng · Mar 2, 2018 · 3d8aa32 · 3d8aa32
1 parent befaceb
commit 3d8aa32
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/packages/transducers/src/xform/flatten-with.ts b/packages/transducers/src/xform/flatten-with.ts
@@ -6,9 +6,9 @@ export function flattenWith<T>(fn: (x: T) => Iterable<T>): Transducer<T | Iterab
     return (rfn: Reducer<any, T>) => {
         const r = rfn[2];
         const flatten = (acc, x) => {
-            x = fn(x);
-            if (x) {
-                for (let y of x) {
+            const xx = fn(x);
+            if (xx) {
+                for (let y of xx) {
                     acc = flatten(acc, y);
                     if (isReduced(acc)) {
                         break;
@@ -18,6 +18,6 @@ export function flattenWith<T>(fn: (x: T) => Iterable<T>): Transducer<T | Iterab
             }
             return r(acc, x);
         };
-        return compR(rfn, fn);
+        return compR(rfn, flatten);
     };
 }