Fix test

dmlc · Feb 21, 2024 · 70b4206 · 70b4206
1 parent 30cf2a0
commit 70b4206
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/test_field_accessor.py b/tests/python/test_field_accessor.py
@@ -60,7 +60,7 @@ def make_tree_stump(left_child_val, right_child_val):
     expected_values = {
         "major_ver": treelite_ver.major,
         "minor_ver": treelite_ver.minor,
-        "patch_ver": treelite_ver.minor,
+        "patch_ver": treelite_ver.micro,
         "threshold_type": 3,  # kFloat64
         "leaf_output_type": 3,  # kFloat64
         "num_tree": 2,
@@ -80,7 +80,7 @@ def make_tree_stump(left_child_val, right_child_val):
         if isinstance(v, np.ndarray):
             np.testing.assert_equal(header.get_field(k), v)
         else:
-            assert header.get_field(k) == v
+            assert header.get_field(k) == v, f"{k=}"
 
     num_tree = header.get_field("num_tree")[0]
     for tree_id in range(num_tree):