Skip to content

Commit

Permalink
update style and lintr
Browse files Browse the repository at this point in the history
  • Loading branch information
Liming Li (李黎明) committed Jun 27, 2024
1 parent ad76e1b commit 1e84b2c
Show file tree
Hide file tree
Showing 4 changed files with 12 additions and 5 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -34,3 +34,4 @@ package.json
RobinCar2.Rcheck/
RobinCar2*.tar.gz
RobinCar2*.tgz
.Rprofile
11 changes: 8 additions & 3 deletions R/treatment_effect.R
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,8 @@ treatment_effect <- function(object, pair, variance, eff_measure, eff_jacobian,
}

#' @export
treatment_effect.prediction_cf <- function(object, pair = names(object), variance = vcovANHECOVA, eff_measure, eff_jacobian, ...) {
treatment_effect.prediction_cf <- function(
object, pair = names(object), variance = vcovANHECOVA, eff_measure, eff_jacobian, ...) {
assert(
test_function(variance),
test_null(variance)
Expand Down Expand Up @@ -54,7 +55,9 @@ treatment_effect.prediction_cf <- function(object, pair = names(object), varianc

#' @export
#' @inheritParams predict_counterfactual
treatment_effect.lm <- function(object, pair = names(object), variance = vcovANHECOVA, eff_measure, eff_jacobian, treatment, data, unbiased = TRUE, ...) {
treatment_effect.lm <- function(
object, pair = names(object), variance = vcovANHECOVA, eff_measure, eff_jacobian,
treatment, data, unbiased = TRUE, ...) {
pc <- predict_counterfactual(object, treatment, data, unbiased)
if (missing(pair)) {
treatment_effect(pc, pair = , , variance = variance, eff_measure = eff_measure, eff_jacobian = eff_jacobian, ...)
Expand All @@ -64,7 +67,9 @@ treatment_effect.lm <- function(object, pair = names(object), variance = vcovANH
}

#' @export
treatment_effect.glm <- function(object, pair, variance = vcovANHECOVA, eff_measure, eff_jacobian, treatment, data = object$data, unbiased = TRUE, ...) {
treatment_effect.glm <- function(
object, pair, variance = vcovANHECOVA, eff_measure, eff_jacobian,
treatment, data = object$data, unbiased = TRUE, ...) {
pc <- predict_counterfactual(object, treatment, data, unbiased)
if (missing(pair)) {
treatment_effect(pc, pair = , , variance = variance, eff_measure = eff_measure, eff_jacobian = eff_jacobian, ...)
Expand Down
2 changes: 1 addition & 1 deletion R/variance_anhecova.R
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
#'
#' @return Named covariance matrix.
#' @export
vcovANHECOVA <- function(x, decompose = TRUE, randomization = "simple", ...) {
vcovANHECOVA <- function(x, decompose = TRUE, randomization = "simple", ...) { # nolint
assert_class(x, "prediction_cf")
assert_flag(decompose)
assert_string(randomization)
Expand Down
3 changes: 2 additions & 1 deletion inst/WORDLIST
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
ANHECOVA
RobinCar
initializer
jacobian
pbo
trt

0 comments on commit 1e84b2c

Please sign in to comment.