tweak: Add TwoNumbersToNumber wrapper to function registry.

2022-09-04 15:09:09 +02:00 · 2022-09-04 15:09:09 +02:00 · 19ddd4eb3f
commit 19ddd4eb3f
parent 462f1c9649
1 changed files with 12 additions and 0 deletions
--- a/packages/squiggle-lang/src/rescript/FunctionRegistry/Library/FR_Number.res
+++ b/packages/squiggle-lang/src/rescript/FunctionRegistry/Library/FR_Number.res
@ -20,6 +20,18 @@ module NumberToNumber = {
    )
 }

+module TwoNumbersToNumber = {
+  let make = (name, fn) =>
+    FnDefinition.make(
+      ~name,
+      ~inputs=[FRTypeNumber, FRTypeNumber],
+      ~run=(_, inputs, _, _) => {
+        inputs->Prepare.ToValueTuple.twoNumbers->E.R2.fmap(fn)->E.R2.fmap(Wrappers.evNumber)
+      },
+      (),
+    )
+}
+
 module ArrayNumberDist = {
  let make = (name, fn) => {
    FnDefinition.make(