quantified.uncertainty/squiggle
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tweak: format

Browse Source
This commit is contained in:
NunoSempere 2022-09-05 16:08:06 +02:00
parent 8f8ca02730
commit 9551aae634
1 changed files with 28 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
packages/squiggle-lang/src/rescript/FunctionRegistry/Library/FR_Danger.res
View File

@ -214,8 +214,9 @@ module Internals = {
let increment = funds /. numDivisions let increment = funds /. numDivisions
let arrayOfIncrements = Belt.Array.makeBy(numDivisionsInt, _ => increment) let arrayOfIncrements = Belt.Array.makeBy(numDivisionsInt, _ => increment)
let findBiggestElementIndex = (xs) => E.A.reducei(xs, 0, (acc, newElement, index) => { let findBiggestElementIndex = xs =>
switch(newElement > xs[acc]){ E.A.reducei(xs, 0, (acc, newElement, index) => {
switch newElement > xs[acc] {
| true => index | true => index
| false => acc | false => acc
} }
@ -231,19 +232,21 @@ module Internals = {
acc, acc,
newIncrement, newIncrement,
) => { ) => {
switch(acc){ switch acc {
| Ok(accInner) => { | Ok(accInner) => {
let oldMarginalReturnsWrapped = accInner.currentMarginalReturns let oldMarginalReturnsWrapped = accInner.currentMarginalReturns
let newAccWrapped = switch(oldMarginalReturnsWrapped){ let newAccWrapped = switch oldMarginalReturnsWrapped {
| Ok(oldMarginalReturns) => { | Ok(oldMarginalReturns) => {
let indexOfBiggestDMR = findBiggestElementIndex(oldMarginalReturns) let indexOfBiggestDMR = findBiggestElementIndex(oldMarginalReturns)
let newOptimalAllocations = Belt.Array.copy(accInner.optimalAllocations) let newOptimalAllocations = Belt.Array.copy(accInner.optimalAllocations)
let newOptimalAllocationsi = let newOptimalAllocationsi = newOptimalAllocations[indexOfBiggestDMR] +. newIncrement
newOptimalAllocations[indexOfBiggestDMR] +. newIncrement
newOptimalAllocations[indexOfBiggestDMR] = newOptimalAllocationsi newOptimalAllocations[indexOfBiggestDMR] = newOptimalAllocationsi
let lambdai = indexOfBiggestDMR == 0 ? lambda1 : lambda2 // to do: generalize let lambdai = indexOfBiggestDMR == 0 ? lambda1 : lambda2 // to do: generalize
let newMarginalResultsLambdai = applyFunctionAtFloatToFloatOption(lambdai, newOptimalAllocationsi) let newMarginalResultsLambdai = applyFunctionAtFloatToFloatOption(
let newCurrentMarginalReturns = switch(newMarginalResultsLambdai){ lambdai,
newOptimalAllocationsi,
)
let newCurrentMarginalReturns = switch newMarginalResultsLambdai {
| Ok(value) => { | Ok(value) => {
let result = Belt.Array.copy(oldMarginalReturns) let result = Belt.Array.copy(oldMarginalReturns)
result[indexOfBiggestDMR] = value result[indexOfBiggestDMR] = value
@ -254,7 +257,7 @@ module Internals = {
let newAcc: diminishingReturnsAccumulatorInner = { let newAcc: diminishingReturnsAccumulatorInner = {
optimalAllocations: newOptimalAllocations, optimalAllocations: newOptimalAllocations,
currentMarginalReturns: newCurrentMarginalReturns currentMarginalReturns: newCurrentMarginalReturns,
} }
Ok(newAcc) Ok(newAcc)
} }