Skip to content

Commit b02b277

Browse files
davidgraeffhcho3
andauthored
tree.h: Fix SetModelPresetVariant variant_index template parameter (#600)
* tree.h: Fix SetModelPresetVariant variant_index template parameter Fixes compiling with -Werror=sign-conversion Fixes #599 * Update include/treelite/tree.h --------- Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
1 parent d0eef62 commit b02b277

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

include/treelite/tree.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -428,8 +428,8 @@ class ModelPreset {
428428

429429
using ModelPresetVariant = std::variant<ModelPreset<float, float>, ModelPreset<double, double>>;
430430

431-
template <int variant_index>
432-
ModelPresetVariant SetModelPresetVariant(int target_variant_index) {
431+
template <std::size_t variant_index>
432+
ModelPresetVariant SetModelPresetVariant(std::size_t target_variant_index) {
433433
ModelPresetVariant result;
434434
if constexpr (variant_index != std::variant_size_v<ModelPresetVariant>) {
435435
if (variant_index == target_variant_index) {

0 commit comments

Comments
 (0)