From ba2cba7ec4aa5f3e3839e46bdfc6bca431e0daf2 Mon Sep 17 00:00:00 2001
From: Hyunsu Cho <chohyu01@cs.washington.edu>
Date: Thu, 22 Feb 2024 00:58:12 -0800
Subject: [PATCH] Require Treelite 4.1.1

---
 cmake/ExternalLibs.cmake | 4 ++--
 ops/conda_env/dev.yml    | 2 +-
 python/pyproject.toml    | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/cmake/ExternalLibs.cmake b/cmake/ExternalLibs.cmake
index 4b8b707..fffa2af 100644
--- a/cmake/ExternalLibs.cmake
+++ b/cmake/ExternalLibs.cmake
@@ -1,7 +1,7 @@
 include(FetchContent)
 
 # Treelite
-find_package(Treelite 4.1.0)
+find_package(Treelite 4.1.1)
 if (Treelite_FOUND)
   set(TREELITE_FROM_SYSTEM_ROOT TRUE)
   set(TREELITE_LIB treelite::treelite)
@@ -10,7 +10,7 @@ else ()
   FetchContent_Declare(
     treelite
     GIT_REPOSITORY https://github.com/dmlc/treelite.git
-    GIT_TAG 4.1.0
+    GIT_TAG 4.1.1
   )
   set(Treelite_BUILD_STATIC_LIBS ON)
   FetchContent_MakeAvailable(treelite)
diff --git a/ops/conda_env/dev.yml b/ops/conda_env/dev.yml
index 7d47176..ab54c3d 100644
--- a/ops/conda_env/dev.yml
+++ b/ops/conda_env/dev.yml
@@ -24,4 +24,4 @@ dependencies:
 - awscli
 - pip
 - python-build
-- treelite=4.1
+- treelite=4.1.1
diff --git a/python/pyproject.toml b/python/pyproject.toml
index 79ff0cd..34625d1 100644
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -29,7 +29,7 @@ dependencies = [
     "numpy",
     "scipy",
     "packaging",
-    "treelite>=4.1"
+    "treelite>=4.1.1"
 ]
 
 [project.urls]