Update annotations.

loophp · Sep 2, 2020 · aa7edf9 · aa7edf9
1 parent ce64bca
commit aa7edf9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/src/Operation/Flip.php b/src/Operation/Flip.php
@@ -15,13 +15,16 @@
  */
 final class Flip extends AbstractOperation
 {
+    /**
+     * @return Closure(Iterator<TKey, T>): Generator<T, TKey>
+     */
     public function __invoke(): Closure
     {
         return
             /**
              * @psalm-param Iterator<TKey, T> $iterator
              *
-             * @psalm-return Generator<int|string, TKey>
+             * @psalm-return Generator<T, TKey>
              */
             static function (Iterator $iterator): Generator {
                 foreach ($iterator as $key => $value) {

diff --git a/src/Operation/Head.php b/src/Operation/Head.php
@@ -16,12 +16,18 @@
 final class Head extends AbstractOperation
 {
     /**
-     * @psalm-return Closure(Iterator<TKey, T>): (Generator<TKey, T>)
+     * @psalm-return Closure(Iterator<TKey, T>): Generator<TKey, T>
      */
     public function __invoke(): Closure
     {
-        return static function (Iterator $iterator): Generator {
-            return yield $iterator->key() => $iterator->current();
-        };
+        return
+            /**
+             * @psalm-param Iterator<TKey, T> $iterator
+             *
+             * @psalm-return Generator<TKey, T>
+             */
+            static function (Iterator $iterator): Generator {
+                return yield $iterator->key() => $iterator->current();
+            };
     }
 }