Fix error message about max depth (#30)

* Fix error message about max depth * Update test APITest.PathTooLong
rapidsai · Dec 22, 2021 · 5af1daa · 5af1daa
1 parent c78fe62
commit 5af1daa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/GPUTreeShap/gpu_treeshap.h b/GPUTreeShap/gpu_treeshap.h
@@ -1087,7 +1087,7 @@ void ValidatePaths(const PathVectorT& device_paths,
                      path_lengths.end(), too_long_op);
 
   if (invalid_length) {
-    throw std::invalid_argument("Tree depth must be <= 32");
+    throw std::invalid_argument("Tree depth must be < 32");
   }
 
   IncorrectVOp<SplitConditionT> incorrect_v_op{device_paths.data().get()};

diff --git a/tests/test_gpu_treeshap.cu b/tests/test_gpu_treeshap.cu
@@ -219,7 +219,7 @@ TEST_F(APITest, PathTooLong) {
   for (size_t i = 1; i < model.size(); i++) {
     model[i] = {0, static_cast<int64_t>(i), 0, {0, 0, 0}, 0, 0};
   }
-  ExpectAPIThrow<std::invalid_argument>("Tree depth must be <= 32");
+  ExpectAPIThrow<std::invalid_argument>("Tree depth must be < 32");
 }
 
 TEST_F(APITest, PathVIncorrect) {