move constants to context + temp schema in test

Author: sfc-gh-bkogan

src/snowflake/snowpark/_internal/udf_utils.py

@@ -59,6 +59,10 @@
 )
 from snowflake.snowpark.types import DataType, StructField, StructType
 from snowflake.snowpark.version import VERSION
+from snowflake.snowpark.context import (
+    _ANACONDA_SHARED_REPOSITORY,
+    _DEFAULT_ARTIFACT_REPOSITORY,
+)
 
 if installed_pandas:
     from snowflake.snowpark.types import (
@@ -1227,11 +1231,6 @@ def resolve_imports_and_packages(
     Optional[str],
     bool,
 ]:
-    from snowflake.snowpark.session import (
-        _ANACONDA_SHARED_REPOSITORY,
-        _DEFAULT_ARTIFACT_REPOSITORY,
-    )
-
     if artifact_repository is None:
         artifact_repository = (
             session._get_default_artifact_repository()

src/snowflake/snowpark/context.py

@@ -48,6 +48,11 @@
 # example: _integral_type_default_precision = {IntegerType: 9}, IntegerType default _precision is 9 now
 _integral_type_default_precision = {}
 
+# The fully qualified name of the Anaconda shared repository (conda channel).
+_ANACONDA_SHARED_REPOSITORY = "snowflake.snowpark.anaconda_shared_repository"
+# In case of failures or the current default artifact repository is unset, we fallback to this
+_DEFAULT_ARTIFACT_REPOSITORY = _ANACONDA_SHARED_REPOSITORY
+
 
 def configure_development_features(
     *,

src/snowflake/snowpark/session.py

@@ -156,6 +156,8 @@
 from snowflake.snowpark.context import (
     _is_execution_environment_sandboxed_for_client,
     _use_scoped_temp_objects,
+    _ANACONDA_SHARED_REPOSITORY,
+    _DEFAULT_ARTIFACT_REPOSITORY,
 )
 from snowflake.snowpark.dataframe import DataFrame
 from snowflake.snowpark.dataframe_reader import DataFrameReader
@@ -321,11 +323,6 @@
 WRITE_PANDAS_CHUNK_SIZE: int = 100000 if is_in_stored_procedure() else None
 WRITE_ARROW_CHUNK_SIZE: int = 100000 if is_in_stored_procedure() else None
 
-# The fully qualified name of the Anaconda shared repository (conda channel).
-_ANACONDA_SHARED_REPOSITORY = "snowflake.snowpark.anaconda_shared_repository"
-# In case of failures or the current default artifact repository is unset, we fallback to this
-_DEFAULT_ARTIFACT_REPOSITORY = _ANACONDA_SHARED_REPOSITORY
-
 
 def _get_active_session() -> "Session":
     with _session_management_lock:

tests/integ/test_packaging.py

@@ -20,7 +20,7 @@
     get_signature,
 )
 from snowflake.snowpark.functions import call_udf, col, count_distinct, sproc, udf
-from snowflake.snowpark.session import _ANACONDA_SHARED_REPOSITORY
+from snowflake.snowpark.context import _ANACONDA_SHARED_REPOSITORY
 from snowflake.snowpark.types import DateType, StringType
 from tests.utils import IS_IN_STORED_PROC, TempObjectType, TestFiles, Utils
 

tests/integ/test_udf.py

@@ -3083,7 +3083,9 @@ def test_urllib() -> str:
 )
 def test_use_default_artifact_repository(db_parameters):
     with Session.builder.configs(db_parameters).create() as session:
-        session.use_schema("public")
+        temp_stage = Utils.random_temp_schema()
+        session.sql(f"create schema {temp_stage}").collect()
+        session.sql(f"use schema {temp_stage}").collect()
         session.sql(
             "ALTER SESSION SET ENABLE_DEFAULT_PYTHON_ARTIFACT_REPOSITORY = true"
         ).collect()
@@ -3115,4 +3117,4 @@ def test_art() -> str:
         finally:
             session._run_query(f"drop function if exists {temp_func_name}(int)")
 
-        session.sql("ALTER schema unset DEFAULT_PYTHON_ARTIFACT_REPOSITORY").collect()
+        session.sql(f"drop schema {temp_stage}").collect()

tests/unit/test_session.py

@@ -28,10 +28,8 @@
     SnowparkInvalidObjectNameException,
     SnowparkSessionException,
 )
-from snowflake.snowpark.session import (
-    _PYTHON_SNOWPARK_USE_SCOPED_TEMP_OBJECTS_STRING,
-    _ANACONDA_SHARED_REPOSITORY,
-)
+from snowflake.snowpark.session import _PYTHON_SNOWPARK_USE_SCOPED_TEMP_OBJECTS_STRING
+from snowflake.snowpark.context import _ANACONDA_SHARED_REPOSITORY
 from snowflake.snowpark.types import StructField, StructType
 
 

tests/unit/test_udf_utils.py

@@ -26,7 +26,7 @@
     resolve_packages_in_client_side_sandbox,
 )
 from snowflake.snowpark._internal.utils import TempObjectType
-from snowflake.snowpark.session import _ANACONDA_SHARED_REPOSITORY
+from snowflake.snowpark.context import _ANACONDA_SHARED_REPOSITORY
 from snowflake.snowpark.types import StringType
 from snowflake.snowpark.version import VERSION