fix: return type of fountain generator

qifi-dev · Oct 11, 2024 · fb622f8 · fb622f8
1 parent f8a4836
commit fb622f8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/packages/generate/src/base-generater.ts b/packages/generate/src/base-generater.ts
@@ -17,7 +17,7 @@ export function createGeneraterWithRender<Result, RenderOptions>(createRender: C
       /**
        * Generate random encoded blocks that **never** ends
        */
-      *fountain(): Generator<Result> {
+      *fountain(): Generator<Result, never> {
         while (true) {
           _fountain ||= encoder.fountain()
           const block = _fountain.next().value

diff --git a/packages/luby-transform/src/encoder.ts b/packages/luby-transform/src/encoder.ts
@@ -46,7 +46,7 @@ export class LtEncoder {
   /**
    * Generate random encoded blocks that **never** ends
    */
-  *fountain(): Generator<EncodedBlock> {
+  *fountain(): Generator<EncodedBlock, never> {
     while (true) {
       const degree = getRandomDegree(this.k)
       const selectedIndices = getRandomIndices(this.k, degree)