chore(sa): fix psalm type infer for Chunk operation

loophp · Dec 11, 2024 · fd1ca32 · fd1ca32
1 parent ce28b5a
commit fd1ca32
Showing 1 changed file with 6 additions and 10 deletions.
diff --git a/src/Operation/Chunk.php b/src/Operation/Chunk.php
@@ -44,19 +44,15 @@ static function (iterable $iterable) use ($sizes): Generator {
                             return;
                         }
 
-                        if (count($chunk) !== $size) {
-                            $chunk[] = $value;
+                        $chunk[] = $value;
 
-                            continue;
-                        }
-
-                        ++$chunkIndex;
+                        if (count($chunk) >= $size) {
+                            ++$chunkIndex;
 
-                        /** @psalm-var non-empty-list $chunk */
+                            yield $chunk;
 
-                        yield $chunk;
-
-                        $chunk = [$value];
+                            $chunk = [];
+                        }
                     }
 
                     if ([] !== $chunk) {