Skip to content

Commit 9da4c59

Browse files
Update exception msg
Co-authored-by: graphite-app[bot] <96075541+graphite-app[bot]@users.noreply.github.com>
1 parent 1d7f7e6 commit 9da4c59

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/snowflake/snowpark/secrets.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -158,13 +158,15 @@ def get_username_password(self, secret_name: str) -> UsernamePassword:
158158
password = self._read_scls_spcs_secret_file(secret_name, "password")
159159
return UsernamePassword(username, password)
160160

161+
161162
def get_cloud_provider_token(self, secret_name: str) -> CloudProviderToken:
162163
# SPCS container currently does not support cloud provider token secrets
163164
raise NotImplementedError(
164-
"Cannot import _snowflake module. get_cloud_provider_token is only supported on Snowflake server environment."
165+
"Cloud provider token secrets are not supported in SPCS container environments."
165166
)
166167

167168

169+
168170
def _is_spcs_environment() -> bool:
169171
return os.getenv(SCLS_SPCS_SECRET_ENV_NAME, None) is not None
170172

0 commit comments

Comments
 (0)