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 .