Fix VerticalDatasetTest

PiperOrigin-RevId: 544655269
google · Jun 30, 2023 · 8bd8961 · 8bd8961
1 parent bb856d8
commit 8bd8961
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/yggdrasil_decision_forests/dataset/BUILD b/yggdrasil_decision_forests/dataset/BUILD
@@ -436,6 +436,7 @@ cc_test(
         "//yggdrasil_decision_forests/utils:filesystem",
         "//yggdrasil_decision_forests/utils:logging",
         "//yggdrasil_decision_forests/utils:test",
+        "//yggdrasil_decision_forests/utils:testing_macros",
     ],
 )
 

diff --git a/yggdrasil_decision_forests/dataset/vertical_dataset_test.cc b/yggdrasil_decision_forests/dataset/vertical_dataset_test.cc
@@ -34,6 +34,7 @@
 #include "yggdrasil_decision_forests/utils/filesystem.h"
 #include "yggdrasil_decision_forests/utils/logging.h"
 #include "yggdrasil_decision_forests/utils/test.h"
+#include "yggdrasil_decision_forests/utils/testing_macros.h"
 
 namespace yggdrasil_decision_forests {
 namespace dataset {