From cdb55f2ea561cc0f95e8deb891d4ea39025a1eda Mon Sep 17 00:00:00 2001
From: pintaoz-aws <167920275+pintaoz-aws@users.noreply.github.com>
Date: Thu, 27 Mar 2025 10:49:59 -0700
Subject: [PATCH] Remove sagemaker from integ test dependency

---
 .github/workflows/pr-checks.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/pr-checks.yml b/.github/workflows/pr-checks.yml
index 0adcf441..79df8d93 100644
--- a/.github/workflows/pr-checks.yml
+++ b/.github/workflows/pr-checks.yml
@@ -66,7 +66,7 @@ jobs:
         run: |
           export PYTHONPATH="$PYTHONPATH:$PWD"
           python -m pip install --upgrade pip
-          pip install scikit-learn sagemaker setuptools awscli
+          pip install scikit-learn setuptools awscli
           pip install -e ".[codegen]"
       - name: Configure AWS Credentials
         uses: aws-actions/configure-aws-credentials@v4
@@ -235,4 +235,4 @@ jobs:
           pylint $python_files --persistent=y --fail-under=8
       - name: Run Black Check
         run: |
-          black --check .
\ No newline at end of file
+          black --check .