fix: types

ricokahler · Apr 14, 2022 · 63dc8e5 · 63dc8e5
1 parent 21a541b
commit 63dc8e5
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/index.d.ts b/index.d.ts
@@ -45,11 +45,11 @@ declare class Lazy<T> implements Iterable<T> {
   static filter<T, R extends T>(
     iterable: Iterable<T>,
     accept: (t: T) => t is R,
-  ): Generator<R>;
+  ): Iterable<R>;
   static filter<T>(
     iterable: Iterable<T>,
     accept: (t: T) => unknown,
-  ): Generator<T>;
+  ): Iterable<T>;
 
   /**
    * Takes in an iterable and returns an iterable generator that yields the
@@ -62,7 +62,10 @@ declare class Lazy<T> implements Iterable<T> {
     depth?: TDepth,
   ): FlatIterable<T, TDepth>;
 
-  static flatMap<T, U>(iterable: T, mapper: (value: T) => U | Iterable<U>): U[];
+  static flatMap<T, R>(
+    iterable: T,
+    mapper: (value: T) => R | Iterable<R>,
+  ): Iterable<R>;
 
   /**
    * Yields the first `n` items of the given iterable and stops further
@@ -219,7 +222,7 @@ declare class Lazy<T> implements Iterable<T> {
     depth?: TDepth,
   ): Lazy<FlatIterable<T, TDepth>>;
 
-  flatMap<U>(mapper: (value: T) => U | Iterable<U>): U[];
+  flatMap<R>(mapper: (value: T) => R | Iterable<R>): Lazy<R>;
 
   /**
    * Yields the first `n` items of the given iterable and stops further