Skip CSF R test on Arm (#1462)

grf-labs · Oct 14, 2024 · c63e0ce · c63e0ce
1 parent a5e5eb2
commit c63e0ce
Showing 1 changed file with 35 additions and 30 deletions.
diff --git a/r-package/grf/tests/testthat/test_causal_survival_forest.R b/r-package/grf/tests/testthat/test_causal_survival_forest.R
@@ -363,34 +363,39 @@ test_that("causal survival forest utility functions are internally consistent",
 # It is done here in addition to ForestCharacterizationTest.cpp as the computation of
 # nuisance components involves a fair amount of work in R.
 test_that("causal survival forest has not changed ", {
-  set.seed(42)
-  n <- 500
-  p <- 5
-  dgp <- "simple1"
-  data <- generate_causal_survival_data(n = n, p = p, dgp = dgp)
-  cs.forest <- causal_survival_forest(round(data$X, 2), round(data$Y, 2), data$W, data$D, horizon = data$Y.max,
-                                      num.trees = 50, seed = 42, num.threads = 4)
-
-  # Update with:
-  # write.table(predict(cs.forest)$predictions, file = "data/causal_survival_oob_predictions.csv", row.names = FALSE, col.names = FALSE)
-  # write.table(predict(cs.forest, round(data$X, 2))$predictions, file = "data/causal_survival_predictions.csv", row.names = FALSE, col.names = FALSE)
-  expected.predictions.oob <- as.numeric(readLines("data/causal_survival_oob_predictions.csv"))
-  expected.predictions <- as.numeric(readLines("data/causal_survival_predictions.csv"))
-
-  expect_equal(predict(cs.forest)$predictions, expected.predictions.oob)
-  expect_equal(predict(cs.forest, round(data$X, 2))$predictions, expected.predictions)
-
-  # With target = "survival.probability"
-  cs.forest.prob <- causal_survival_forest(round(data$X, 2), round(data$Y, 2), data$W, data$D,
-                                           target = "survival.probability", horizon = 0.5,
-                                           num.trees = 50, seed = 42, num.threads = 4)
-
-  # Update with:
-  # write.table(predict(cs.forest.prob)$predictions, file = "data/causal_survival_oob_predictions_prob.csv", row.names = FALSE, col.names = FALSE)
-  # write.table(predict(cs.forest.prob, round(data$X, 2))$predictions, file = "data/causal_survival_predictions_prob.csv", row.names = FALSE, col.names = FALSE)
-  expected.predictions.oob.prob <- as.numeric(readLines("data/causal_survival_oob_predictions_prob.csv"))
-  expected.predictions.prob <- as.numeric(readLines("data/causal_survival_predictions_prob.csv"))
-
-  expect_equal(predict(cs.forest.prob)$predictions, expected.predictions.oob.prob)
-  expect_equal(predict(cs.forest.prob, round(data$X, 2))$predictions, expected.predictions.prob)
+  # Skip if running on Apple silicon
+  if (R.version$arch == "aarch64") {
+    expect_equal(1, 1)
+  } else {
+    set.seed(42)
+    n <- 500
+    p <- 5
+    dgp <- "simple1"
+    data <- generate_causal_survival_data(n = n, p = p, dgp = dgp)
+    cs.forest <- causal_survival_forest(round(data$X, 2), round(data$Y, 2), data$W, data$D, horizon = data$Y.max,
+                                        num.trees = 50, seed = 42, num.threads = 4)
+
+    # Update with:
+    # write.table(predict(cs.forest)$predictions, file = "data/causal_survival_oob_predictions.csv", row.names = FALSE, col.names = FALSE)
+    # write.table(predict(cs.forest, round(data$X, 2))$predictions, file = "data/causal_survival_predictions.csv", row.names = FALSE, col.names = FALSE)
+    expected.predictions.oob <- as.numeric(readLines("data/causal_survival_oob_predictions.csv"))
+    expected.predictions <- as.numeric(readLines("data/causal_survival_predictions.csv"))
+
+    expect_equal(predict(cs.forest)$predictions, expected.predictions.oob)
+    expect_equal(predict(cs.forest, round(data$X, 2))$predictions, expected.predictions)
+
+    # With target = "survival.probability"
+    cs.forest.prob <- causal_survival_forest(round(data$X, 2), round(data$Y, 2), data$W, data$D,
+                                            target = "survival.probability", horizon = 0.5,
+                                            num.trees = 50, seed = 42, num.threads = 4)
+
+    # Update with:
+    # write.table(predict(cs.forest.prob)$predictions, file = "data/causal_survival_oob_predictions_prob.csv", row.names = FALSE, col.names = FALSE)
+    # write.table(predict(cs.forest.prob, round(data$X, 2))$predictions, file = "data/causal_survival_predictions_prob.csv", row.names = FALSE, col.names = FALSE)
+    expected.predictions.oob.prob <- as.numeric(readLines("data/causal_survival_oob_predictions_prob.csv"))
+    expected.predictions.prob <- as.numeric(readLines("data/causal_survival_predictions_prob.csv"))
+
+    expect_equal(predict(cs.forest.prob)$predictions, expected.predictions.oob.prob)
+    expect_equal(predict(cs.forest.prob, round(data$X, 2))$predictions, expected.predictions.prob)
+  }
 })