up tmle

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"

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)