fix(transducers): update arg types for choices() & weightedRandom()

thi-ng · Sep 17, 2018 · eb67426 · eb67426
1 parent bd22811
commit eb67426
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/packages/transducers/src/func/weighted-random.ts b/packages/transducers/src/func/weighted-random.ts
@@ -10,7 +10,7 @@ import { tuples } from "../iter/tuples";
  * @param choices
  * @param weights
  */
-export function weightedRandom<T>(choices: T[], weights?: number[]) {
+export function weightedRandom<T>(choices: ArrayLike<T> & Iterable<T>, weights?: ArrayLike<number> & Iterable<number>) {
     const n = choices.length;
     const opts = [...tuples(choices, weights || repeat(1))].sort((a, b) => b[1] - a[1]);
     let total = 0, i, r, sum;

diff --git a/packages/transducers/src/iter/choices.ts b/packages/transducers/src/iter/choices.ts
@@ -2,9 +2,9 @@ import { weightedRandom } from "../func/weighted-random";
 import { repeatedly } from "./repeatedly";
 
 /**
- * Returns an infinite iterator of random choices and their (optional) weights.
- * If `weights` are given, it must be the same size as `choices`. If omitted,
- * each choice will have same probability.
+ * Returns an infinite iterator of random choices and their (optional)
+ * weights. If `weights` is given, it must have at least the same size
+ * as `choices`. If omitted, each choice will have same probability.
  *
  * See: `weightedRandom()`
  *
@@ -16,7 +16,7 @@ import { repeatedly } from "./repeatedly";
  * @param choices
  * @param weights
  */
-export function choices<T>(choices: T[], weights?: number[]) {
+export function choices<T>(choices: ArrayLike<T> & Iterable<T>, weights?: ArrayLike<number> & Iterable<number>) {
     return repeatedly(
         weights ?
             weightedRandom(choices, weights) :