From 5265ff9a96c20f63a894dd440b8b1279ad4557f1 Mon Sep 17 00:00:00 2001
From: Simon Zhao <simonyansenzhao@gmail.com>
Date: Tue, 14 Jan 2025 15:56:38 +0800
Subject: [PATCH] Use cornac 2.3.0 for Python 3.9+

Signed-off-by: Simon Zhao <simonyansenzhao@gmail.com>
---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 41b4c8aed..98b6b9b39 100644
--- a/setup.py
+++ b/setup.py
@@ -28,7 +28,8 @@
 
 install_requires = [
     "category-encoders>=2.6.0,<3",  # requires packaging
-    "cornac>=1.15.2,<3",  # requires packaging, tqdm
+    "cornac>=1.15.2,<=2.2.2;python_version<='3.8'",
+    "cornac>=2.3.0,<3;python_version>='3.9'",  # requires packaging, tqdm
     "hyperopt>=0.2.7,<1",
     "lightgbm>=4.0.0,<5",
     "locust>=2.12.2,<3",  # requires jinja2