change implementation

bpolaszek · Feb 4, 2021 · 7e8f271 · 7e8f271
1 parent 60dd046
commit 7e8f271
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/iterable-functions.php b/src/iterable-functions.php
@@ -10,7 +10,6 @@
 use Traversable;
 
 use function array_filter;
-use function array_map;
 use function array_values;
 use function iterator_to_array;
 
@@ -23,11 +22,9 @@
  */
 function iterable_map(iterable $iterable, callable $map): iterable
 {
-    if ($iterable instanceof Traversable) {
-        return new ArrayIterator(array_map($map, iterator_to_array($iterable)));
+    foreach ($iterable as $key => $item) {
+        yield $key => $map($item);
     }
-
-    return array_map($map, $iterable);
 }
 
 /**