diff --git a/Project.toml b/Project.toml index 9aa819a..6a8f8c6 100644 --- a/Project.toml +++ b/Project.toml @@ -55,7 +55,7 @@ MLJModels = "0.16" MLJXGBoostInterface = "0.3.4" MultipleTesting = "0.6.0" Optim = "1.7" -TMLE = "0.14.0" +TMLE = "0.15.0" Tables = "1.10.1" YAML = "0.4.9" julia = "1.7, 1" diff --git a/src/utils.jl b/src/utils.jl index 0f46498..bab266c 100644 --- a/src/utils.jl +++ b/src/utils.jl @@ -82,14 +82,14 @@ This explicitely requires that the following columns belong to the dataset: All ATE parameters are generated. """ -function TMLE.factorialATE(dataset) +function factorialATE(dataset) colnames = names(dataset) "T" ∈ colnames || throw(ArgumentError("No column 'T' found in the dataset for the treatment variable.")) "Y" ∈ colnames || throw(ArgumentError("No column 'Y' found in the dataset for the outcome variable.")) confounding_variables = Tuple(name for name in colnames if occursin(r"^W", name)) length(confounding_variables) > 0 || throw(ArgumentError("Could not find any confounding variable (starting with 'W') in the dataset.")) - return [factorialATE(dataset, (:T, ), :Y; confounders=confounding_variables)] + return [factorialEstimand(ATE, dataset, (:T, ), :Y; confounders=confounding_variables)] end instantiate_config(file::AbstractString) = read_estimands_config(file)