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

Minor renaming

Browse Source
This commit is contained in:
Ozzie Gooen 2020-07-02 14:33:41 +01:00
parent 96df9ced85
commit 19e9eaee83
1 changed files with 10 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/distPlus/symbolic/TreeNode.re
View File

@ -267,11 +267,11 @@ module TreeNode = {
}; };
module FloatFromDist = { module FloatFromDist = {
let evaluateFromSymbolic = (distToFloatOp: distToFloatOperation, s) => { let symbolicToLeaf = (distToFloatOp: distToFloatOperation, s) => {
SymbolicDist.T.operate(distToFloatOp, s) SymbolicDist.T.operate(distToFloatOp, s)
|> E.R.bind(_, v => Ok(`Leaf(`SymbolicDist(`Float(v))))); |> E.R.bind(_, v => Ok(`Leaf(`SymbolicDist(`Float(v)))));
}; };
let evaluateFromRenderedDist = let renderedToLeaf =
(distToFloatOp: distToFloatOperation, rs: DistTypes.shape) (distToFloatOp: distToFloatOperation, rs: DistTypes.shape)
: result(treeNode, string) => { : result(treeNode, string) => {
Distributions.Shape.operate(distToFloatOp, rs) Distributions.Shape.operate(distToFloatOp, rs)
@ -285,9 +285,8 @@ module TreeNode = {
) )
: result(treeNode, string) => { : result(treeNode, string) => {
switch (t) { switch (t) {
| `Leaf(`SymbolicDist(s)) => evaluateFromSymbolic(distToFloatOp, s) // we want to evaluate the distToFloatOp on the symbolic dist | `Leaf(`SymbolicDist(s)) => symbolicToLeaf(distToFloatOp, s) // we want to evaluate the distToFloatOp on the symbolic dist
| `Leaf(`RenderedDist(rs)) => | `Leaf(`RenderedDist(rs)) => renderedToLeaf(distToFloatOp, rs)
evaluateFromRenderedDist(distToFloatOp, rs)
| `Operation(op) => | `Operation(op) =>
E.R.bind( E.R.bind(
operationToLeaf(op), operationToLeaf(op),
@ -310,10 +309,7 @@ module TreeNode = {
Ok(`Leaf(`RenderedDist(SymbolicDist.T.toShape(sampleCount, d)))) Ok(`Leaf(`RenderedDist(SymbolicDist.T.toShape(sampleCount, d))))
| `Leaf(`RenderedDist(_)) as t => Ok(t) // already a rendered shape, we're done here | `Leaf(`RenderedDist(_)) as t => Ok(t) // already a rendered shape, we're done here
| `Operation(op) => | `Operation(op) =>
E.R.bind( E.R.bind(operationToLeaf(op), toLeaf(operationToLeaf, sampleCount))
operationToLeaf(op),
toLeaf(operationToLeaf, sampleCount),
)
}; };
}; };
}; };
@ -355,11 +351,7 @@ module TreeNode = {
FloatFromDist.toLeaf(operationToLeaf(sampleCount), distToFloatOp, t) FloatFromDist.toLeaf(operationToLeaf(sampleCount), distToFloatOp, t)
| `Normalize(t) => Normalize.toLeaf(operationToLeaf(sampleCount), t) | `Normalize(t) => Normalize.toLeaf(operationToLeaf(sampleCount), t)
| `Render(t) => | `Render(t) =>
Render.toLeaf( Render.toLeaf(operationToLeaf(sampleCount), sampleCount, t)
operationToLeaf(sampleCount),
sampleCount,
t,
)
}; };
}; };