From 9aa8789013831b5f509e715c6908f25a7fda39b5 Mon Sep 17 00:00:00 2001 From: Ruinong Tian Date: Thu, 30 Jan 2025 22:02:59 +0000 Subject: [PATCH] restrict catboost to cpu version --- build_artifacts/v2/v2.2/v2.2.2/cpu.env.in | 1 + 1 file changed, 1 insertion(+) diff --git a/build_artifacts/v2/v2.2/v2.2.2/cpu.env.in b/build_artifacts/v2/v2.2/v2.2.2/cpu.env.in index d80a0a0e..387938c9 100644 --- a/build_artifacts/v2/v2.2/v2.2.2/cpu.env.in +++ b/build_artifacts/v2/v2.2/v2.2.2/cpu.env.in @@ -34,6 +34,7 @@ conda-forge::boto3[version='>=1.34.162,<1.35.0'] conda-forge::sagemaker-python-sdk[version='>=2.227.0,<2.228.0'] conda-forge::supervisor[version='>=4.2.5,<4.3.0'] conda-forge::autogluon[version='>=1.1.1,<1.2.0'] +conda-forge::catboost[version='>=1.1.1,<1.2.0',build='*cpu*'] conda-forge::aws-glue-sessions[version='>=1.0.7,<1.1.0'] conda-forge::sagemaker-kernel-wrapper[version='>=0.0.4,<0.1.0'] conda-forge::jupyter-ai[version='>=2.28.3,<2.29.0']