diff --git a/shared_python_ci_env.yml b/shared_python_ci_env.yml
index 4de72b7..40348bd 100644
--- a/shared_python_ci_env.yml
+++ b/shared_python_ci_env.yml
@@ -12,4 +12,5 @@ dependencies:
       - snowflake-cli-labs>=2.0.0
       - pytest
       - streamlit>=1.26.0
+      - snowflake-ml-python