Skip to content

Commit

Permalink
Minor change to the testing utils
Browse files Browse the repository at this point in the history
  • Loading branch information
weizhangstats committed Apr 5, 2024
1 parent f5e239f commit 5da41bd
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion inst/AD_test_utils.R
Original file line number Diff line number Diff line change
Expand Up @@ -1255,7 +1255,7 @@ setup_update_and_constant_nodes_for_tests <- function(model,
## derivNodes <- model$expandNodeNames(derivNodes) # do not do this because do not want vector node names
nNodes <- length(derivNodes)
calcNodes <- model$getDependencies(derivNodes)
ucNodes <- makeDerivsInfo(model, derivNodes, calcNodes, dataAsConstantNodes = TRUE)
ucNodes <- makeModelDerivsInfo(model, derivNodes, calcNodes, dataAsConstantNodes = TRUE)
updateNodes <- ucNodes$updateNodes
constantNodes <- ucNodes$constantNodes
updateNodes <- setdiff(updateNodes, forceConstantNodes) # remove forceConstants from updates
Expand Down
8 changes: 4 additions & 4 deletions inst/test_utils.R
Original file line number Diff line number Diff line change
Expand Up @@ -1314,7 +1314,7 @@ derivsNimbleFunctionMeta <- nimbleFunction(
setup = function(model, calcNodes, wrt, reset = FALSE) {
innerWrtVec <- seq_along(model$expandNodeNames(wrt, returnScalarComponents = TRUE))
d <- length(innerWrtVec)
derivsInfo <- makeDerivsInfo(model, wrt, calcNodes)
derivsInfo <- makeModelDerivsInfo(model, wrt, calcNodes)
updateNodes <- derivsInfo$updateNodes
constantNodes <- derivsInfo$constantNodes
},
Expand Down Expand Up @@ -1375,7 +1375,7 @@ derivsNimbleFunctionParamTransform <- nimbleFunction(
my_parameterTransform <- parameterTransform(model, wrtNodesAsScalars)
d <- my_parameterTransform$getTransformedLength()
nimDerivs_wrt <- 1:d
derivsInfo <- makeDerivsInfo(model, wrt, calcNodes)
derivsInfo <- makeModelDerivsInfo(model, wrt, calcNodes)
updateNodes <- derivsInfo$updateNodes
constantNodes <- derivsInfo$constantNodes
},
Expand Down Expand Up @@ -1407,7 +1407,7 @@ derivsNimbleFunctionParamTransformMeta <- nimbleFunction(
my_parameterTransform <- parameterTransform(model, wrtNodesAsScalars)
d <- my_parameterTransform$getTransformedLength()
nimDerivs_wrt <- 1:d
derivsInfo <- makeDerivsInfo(model, wrt, calcNodes)
derivsInfo <- makeModelDerivsInfo(model, wrt, calcNodes)
updateNodes <- derivsInfo$updateNodes
constantNodes <- derivsInfo$constantNodes
},
Expand Down Expand Up @@ -1811,7 +1811,7 @@ test_ADModelCalculate_internal <- function(model, name = 'unknown', xOrig = NULL
if(!is.null(d) && length(d) == 2 && d[1] == d[2]) return(TRUE) else return(FALSE)
}

derivsInfo <- makeDerivsInfo(model, wrt, calcNodes)
derivsInfo <- makeModelDerivsInfo(model, wrt, calcNodes)
updateNodes <- derivsInfo$updateNodes
constantNodes <- derivsInfo$constantNodes

Expand Down

0 comments on commit 5da41bd

Please sign in to comment.