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: