From 6eb680f1b77fb35b0966de1aa4a1b187671a525a Mon Sep 17 00:00:00 2001 From: Bruce Hill Date: Tue, 2 Apr 2024 23:40:18 -0400 Subject: Add array:sample(n, weights) --- compile.c | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/compile.c b/compile.c index 08e2fe48..335134e8 100644 --- a/compile.c +++ b/compile.c @@ -1484,7 +1484,8 @@ CORD compile(env_t *env, ast_t *ast) } else if (streq(call->name, "sample")) { CORD self = compile_to_pointer_depth(env, call->self, 0, false); arg_t *arg_spec = new(arg_t, .name="count", .type=Type(IntType, .bits=64), - .next=new(arg_t, .name="weights", .type=self_value_t, .default_val=FakeAST(Array))); + .next=new(arg_t, .name="weights", .type=Type(ArrayType, .item_type=Type(NumType, .bits=64)), + .default_val=FakeAST(Array, .type=new(type_ast_t, .tag=VarTypeAST, .__data.VarTypeAST.name="Num")))); return CORD_all("Array$sample(", self, ", ", compile_arguments(env, ast, arg_spec, call->args), ", ", compile_type_info(env, self_value_t), ")"); } else if (streq(call->name, "shuffle")) { -- cgit v1.2.3