From abc7be3b866f31ff7256660de97779df1a4e64d0 Mon Sep 17 00:00:00 2001
From: Marek Novotny <marek.novotny@h2o.ai>
Date: Tue, 19 Sep 2023 17:05:18 +0200
Subject: [PATCH] Upgrade causalml

---
 h2o-py/test-requirements.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/h2o-py/test-requirements.txt b/h2o-py/test-requirements.txt
index 20eea5fe0cb7..fb38fa7be0d0 100644
--- a/h2o-py/test-requirements.txt
+++ b/h2o-py/test-requirements.txt
@@ -31,8 +31,8 @@ shap==0.29.1
 boto3==1.18.65
 lifelines>=0.19.5
 xlsxwriter==1.3.7
-cython==0.29.34
-causalml==0.12.3
+causalml==0.12.3; python_version <= '3.9'
+causalml==0.14.1; python_version > '3.9'
 colorama
 # not directly required, pinned by Snyk to avoid a vulnerability
 pygments==2.7.4; python_version <= '3.7'