From 70b42063fe14410b1cf75bcf4e0049faec1a3888 Mon Sep 17 00:00:00 2001 From: Hyunsu Cho Date: Wed, 21 Feb 2024 14:06:39 -0800 Subject: [PATCH] Fix test --- tests/python/test_field_accessor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/python/test_field_accessor.py b/tests/python/test_field_accessor.py index 5b9f7770..d0108272 100644 --- 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):