From 1418bbe58b86691ae6214785899e7d14adbaa10b Mon Sep 17 00:00:00 2001
From: nabenabe0928 <shuhei.watanabe.utokyo@gmail.com>
Date: Fri, 20 Dec 2024 03:53:22 +0100
Subject: [PATCH] Fix CI by adding a version constraint

---
 .github/workflows/checks-and-tests.yml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/checks-and-tests.yml b/.github/workflows/checks-and-tests.yml
index 48e3a306..b86069e6 100644
--- a/.github/workflows/checks-and-tests.yml
+++ b/.github/workflows/checks-and-tests.yml
@@ -189,7 +189,8 @@ jobs:
     uses: ./.github/workflows/checks_template.yml
     with:
       integration_name: 'skorch'
-      extra_cmds: pip install torch
+      # TODO: Remove the version constraint once https://github.com/uber/causalml/issues/808 is resolved.
+      extra_cmds: pip install torch "scikit-learn<1.6.0"
       deprecated: false
 
   tensorboard: