diff --git a/CHANGELOG.md b/CHANGELOG.md index bbbf449201..8a7425cf32 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,19 +1,6 @@ # Release History -## 1.47.0 (TBD) - -### Snowpark Python API Updates - -#### New Features - -#### Bug Fixes - -- Fixed a bug where `Session.udf.register_from_file` did not properly process the `strict` and `secure` parameters. - -#### Improvements - - -## 1.46.0 (TBD) +## 1.46.0 (2026-02-25) ### Snowpark Python API Updates @@ -26,6 +13,7 @@ - Fixed a bug where `cloudpickle` was not automatically added to the package list when using `artifact_repository` with custom packages, causing `ModuleNotFoundError` at runtime. - Fixed a bug when reading xml with custom schema, result include element attributes when column is not `StructType` type. +- Fixed a bug where `Session.udf.register_from_file` did not properly process the `strict` and `secure` parameters. #### Improvements diff --git a/recipe/meta.yaml b/recipe/meta.yaml index e85a0dad44..6437286f05 100644 --- a/recipe/meta.yaml +++ b/recipe/meta.yaml @@ -1,5 +1,5 @@ {% set name = "snowflake-snowpark-python" %} -{% set version = "1.45.0" %} +{% set version = "1.46.0" %} {% set noarch_build = (os.environ.get('SNOWFLAKE_SNOWPARK_PYTHON_NOARCH_BUILD', 'false')) == 'true' %} {% set build_number = os.environ.get('SNOWFLAKE_SNOWPARK_PYTHON_BUILD_NUMBER', 0) %} diff --git a/src/snowflake/snowpark/_internal/udf_utils.py b/src/snowflake/snowpark/_internal/udf_utils.py index 7c0f9e01f4..f33de2311f 100644 --- a/src/snowflake/snowpark/_internal/udf_utils.py +++ b/src/snowflake/snowpark/_internal/udf_utils.py @@ -1142,9 +1142,9 @@ def add_snowpark_package_to_sproc_packages( packages = [this_package] else: with session._package_lock: - existing_packages = session._artifact_repository_packages[ + existing_packages = session._get_packages_by_artifact_repository( artifact_repository - ] + ) if package_name not in existing_packages: packages = list(existing_packages.values()) + [this_package] return packages @@ -1239,7 +1239,9 @@ def resolve_imports_and_packages( ) existing_packages_dict = ( - session._artifact_repository_packages[artifact_repository] if session else {} + session._get_packages_by_artifact_repository(artifact_repository) + if session + else {} ) if artifact_repository != _ANACONDA_SHARED_REPOSITORY: @@ -1248,7 +1250,9 @@ def resolve_imports_and_packages( if not packages and session: resolved_packages = list( session._resolve_packages( - [], artifact_repository, existing_packages_dict + [], + artifact_repository=artifact_repository, + existing_packages_dict=existing_packages_dict, ) ) elif packages: @@ -1286,8 +1290,8 @@ def resolve_imports_and_packages( resolved_packages = ( session._resolve_packages( packages, - artifact_repository, - {}, # ignore session packages if passed in explicitly + artifact_repository=artifact_repository, + existing_packages_dict={}, # ignore session packages if passed in explicitly include_pandas=is_pandas_udf, statement_params=statement_params, _suppress_local_package_warnings=_suppress_local_package_warnings, @@ -1295,8 +1299,8 @@ def resolve_imports_and_packages( if packages is not None else session._resolve_packages( [], - artifact_repository, - existing_packages_dict, + artifact_repository=artifact_repository, + existing_packages_dict=existing_packages_dict, validate_package=False, include_pandas=is_pandas_udf, statement_params=statement_params, diff --git a/src/snowflake/snowpark/session.py b/src/snowflake/snowpark/session.py index ad583da1a7..80f7a292d4 100644 --- a/src/snowflake/snowpark/session.py +++ b/src/snowflake/snowpark/session.py @@ -601,6 +601,9 @@ def __init__( self._conn = conn self._query_tag = None self._import_paths: Dict[str, Tuple[Optional[str], Optional[str]]] = {} + # packages under the DEFAULT_ARTIFACT_REPOSITORY + # due to server side accessing private session members, this cannot be merged with _artifact_repository_packages + self._packages: Dict[str, str] = {} # map of artifact repository name -> packages that should be added to functions under that repository self._artifact_repository_packages: DefaultDict[ str, Dict[str, str] @@ -1599,6 +1602,14 @@ def _list_files_in_stage( prefix_length = get_stage_file_prefix_length(stage_location) return {str(row[0])[prefix_length:] for row in file_list} + def _get_packages_by_artifact_repository( + self, artifact_repository: str + ) -> Dict[str, str]: + if artifact_repository == _DEFAULT_ARTIFACT_REPOSITORY: + return self._packages + else: + return self._artifact_repository_packages[artifact_repository] + def get_packages(self, artifact_repository: Optional[str] = None) -> Dict[str, str]: """ Returns a ``dict`` of packages added for user-defined functions (UDFs). @@ -1613,7 +1624,7 @@ def get_packages(self, artifact_repository: Optional[str] = None) -> Dict[str, s artifact_repository = self._get_default_artifact_repository() with self._package_lock: - return self._artifact_repository_packages[artifact_repository].copy() + return self._get_packages_by_artifact_repository(artifact_repository).copy() def add_packages( self, @@ -1686,8 +1697,10 @@ def add_packages( self._resolve_packages( parse_positional_args_to_list(*packages), - artifact_repository, - self._artifact_repository_packages[artifact_repository], + artifact_repository=artifact_repository, + existing_packages_dict=self._get_packages_by_artifact_repository( + artifact_repository + ), ) def remove_package( @@ -1724,7 +1737,7 @@ def remove_package( artifact_repository = self._get_default_artifact_repository() with self._package_lock: - packages = self._artifact_repository_packages[artifact_repository] + packages = self._get_packages_by_artifact_repository(artifact_repository) if package_name in packages: packages.pop(package_name) else: @@ -1742,7 +1755,7 @@ def clear_packages( artifact_repository = self._get_default_artifact_repository() with self._package_lock: - self._artifact_repository_packages[artifact_repository].clear() + self._get_packages_by_artifact_repository(artifact_repository).clear() def add_requirements( self, @@ -2110,11 +2123,11 @@ def _get_req_identifiers_list( def _resolve_packages( self, packages: List[Union[str, ModuleType]], - artifact_repository: str, - existing_packages_dict: Dict[str, str], + existing_packages_dict: Dict[str, str] = None, validate_package: bool = True, include_pandas: bool = False, statement_params: Optional[Dict[str, str]] = None, + artifact_repository: str = None, **kwargs, ) -> List[str]: """ @@ -2132,6 +2145,13 @@ def _resolve_packages( Returns: List[str]: List of package specifiers """ + if artifact_repository is None: + artifact_repository = self._get_default_artifact_repository() + if existing_packages_dict is None: + existing_packages_dict = self._get_packages_by_artifact_repository( + artifact_repository + ) + # Always include cloudpickle extra_modules = [cloudpickle] if include_pandas: @@ -2404,7 +2424,10 @@ def _get_default_artifact_repository(self) -> str: if isinstance(self._conn, MockServerConnection): return _DEFAULT_ARTIFACT_REPOSITORY - cache_key = (self.get_current_database(), self.get_current_schema()) + account = self.get_current_account() + database = self.get_current_database() + schema = self.get_current_schema() + cache_key = (database, schema) if ( self._default_artifact_repository_cache is not None @@ -2414,8 +2437,15 @@ def _get_default_artifact_repository(self) -> str: try: python_version = f"{sys.version_info.major}.{sys.version_info.minor}" + entity_selector_args = ( + f"'schema', '{schema}'" + if schema + else f"'database', '{database}'" + if database + else f"'account', '{account}'" + ) result = self._run_query( - f"SELECT SYSTEM$GET_DEFAULT_PYTHON_ARTIFACT_REPOSITORY('{python_version}')" + f"SELECT SYSTEM$GET_DEFAULT_PYTHON_ARTIFACT_REPOSITORY('{python_version}', {entity_selector_args})" ) value = result[0][0] if result else None resolved = value or _DEFAULT_ARTIFACT_REPOSITORY diff --git a/src/snowflake/snowpark/version.py b/src/snowflake/snowpark/version.py index 18bceb8cc8..825885bb48 100644 --- a/src/snowflake/snowpark/version.py +++ b/src/snowflake/snowpark/version.py @@ -2,4 +2,4 @@ # Update this for the versions -VERSION = (1, 45, 0) +VERSION = (1, 46, 0) diff --git a/tests/ast/data/DataFrame.agg.test b/tests/ast/data/DataFrame.agg.test index eeaf72335c..3dde66874a 100644 --- a/tests/ast/data/DataFrame.agg.test +++ b/tests/ast/data/DataFrame.agg.test @@ -508,6 +508,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.ai.test b/tests/ast/data/DataFrame.ai.test index 69d9c81b8d..fe3239a9e4 100644 --- a/tests/ast/data/DataFrame.ai.test +++ b/tests/ast/data/DataFrame.ai.test @@ -2146,6 +2146,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.col_ilike.test b/tests/ast/data/DataFrame.col_ilike.test index 0ee6c3b9a0..3fb638fc17 100644 --- a/tests/ast/data/DataFrame.col_ilike.test +++ b/tests/ast/data/DataFrame.col_ilike.test @@ -239,6 +239,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.collect.test b/tests/ast/data/DataFrame.collect.test index 2ffad9af14..b1c607b26d 100644 --- a/tests/ast/data/DataFrame.collect.test +++ b/tests/ast/data/DataFrame.collect.test @@ -511,6 +511,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.count.test b/tests/ast/data/DataFrame.count.test index 4714eb3bf2..07ea86d225 100644 --- a/tests/ast/data/DataFrame.count.test +++ b/tests/ast/data/DataFrame.count.test @@ -228,6 +228,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.count2.test b/tests/ast/data/DataFrame.count2.test index 2d17956d29..88c11dc97b 100644 --- a/tests/ast/data/DataFrame.count2.test +++ b/tests/ast/data/DataFrame.count2.test @@ -315,6 +315,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.create_or_replace.test b/tests/ast/data/DataFrame.create_or_replace.test index 10b64ff5c2..e72c759e64 100644 --- a/tests/ast/data/DataFrame.create_or_replace.test +++ b/tests/ast/data/DataFrame.create_or_replace.test @@ -756,6 +756,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.cross_join.lsuffix.test b/tests/ast/data/DataFrame.cross_join.lsuffix.test index 95439cc3a0..8b47ec1916 100644 --- a/tests/ast/data/DataFrame.cross_join.lsuffix.test +++ b/tests/ast/data/DataFrame.cross_join.lsuffix.test @@ -169,6 +169,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.cross_join.rsuffix.test b/tests/ast/data/DataFrame.cross_join.rsuffix.test index f3ac47c7ee..04fa6fb1ec 100644 --- a/tests/ast/data/DataFrame.cross_join.rsuffix.test +++ b/tests/ast/data/DataFrame.cross_join.rsuffix.test @@ -169,6 +169,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.cross_join.suffix.test b/tests/ast/data/DataFrame.cross_join.suffix.test index 63dbb9de4f..e9042a3c97 100644 --- a/tests/ast/data/DataFrame.cross_join.suffix.test +++ b/tests/ast/data/DataFrame.cross_join.suffix.test @@ -172,6 +172,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.describe.test b/tests/ast/data/DataFrame.describe.test index 61c5981ed9..13b17035ef 100644 --- a/tests/ast/data/DataFrame.describe.test +++ b/tests/ast/data/DataFrame.describe.test @@ -192,6 +192,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.flatten.test b/tests/ast/data/DataFrame.flatten.test index d12847f08d..02b44be8a5 100644 --- a/tests/ast/data/DataFrame.flatten.test +++ b/tests/ast/data/DataFrame.flatten.test @@ -175,6 +175,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.indexers.test b/tests/ast/data/DataFrame.indexers.test index ed0ad9e1cf..c6a2830a72 100644 --- a/tests/ast/data/DataFrame.indexers.test +++ b/tests/ast/data/DataFrame.indexers.test @@ -207,6 +207,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.join.inner.column.test b/tests/ast/data/DataFrame.join.inner.column.test index 849fd7a727..c2297ddc71 100644 --- a/tests/ast/data/DataFrame.join.inner.column.test +++ b/tests/ast/data/DataFrame.join.inner.column.test @@ -239,6 +239,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.join.inner.column_list.test b/tests/ast/data/DataFrame.join.inner.column_list.test index fae89e19c5..13119f1a82 100644 --- a/tests/ast/data/DataFrame.join.inner.column_list.test +++ b/tests/ast/data/DataFrame.join.inner.column_list.test @@ -201,6 +201,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.join.inner.column_list_predicate.test b/tests/ast/data/DataFrame.join.inner.column_list_predicate.test index 70b1b5c7cb..a4aa4656cf 100644 --- a/tests/ast/data/DataFrame.join.inner.column_list_predicate.test +++ b/tests/ast/data/DataFrame.join.inner.column_list_predicate.test @@ -326,6 +326,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.join.inner.predicate.test b/tests/ast/data/DataFrame.join.inner.predicate.test index 42ea3a29f6..7ee58ee36d 100644 --- a/tests/ast/data/DataFrame.join.inner.predicate.test +++ b/tests/ast/data/DataFrame.join.inner.predicate.test @@ -287,6 +287,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.join.inner.predicate_rsuffix.test b/tests/ast/data/DataFrame.join.inner.predicate_rsuffix.test index ca585c3f5a..9b14da41d1 100644 --- a/tests/ast/data/DataFrame.join.inner.predicate_rsuffix.test +++ b/tests/ast/data/DataFrame.join.inner.predicate_rsuffix.test @@ -277,6 +277,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.join.left_outer.column.test b/tests/ast/data/DataFrame.join.left_outer.column.test index 8c3dec8978..3c5d95fc53 100644 --- a/tests/ast/data/DataFrame.join.left_outer.column.test +++ b/tests/ast/data/DataFrame.join.left_outer.column.test @@ -201,6 +201,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.join.right_outer.predicate.test b/tests/ast/data/DataFrame.join.right_outer.predicate.test index 7546a6dae4..693216a7bc 100644 --- a/tests/ast/data/DataFrame.join.right_outer.predicate.test +++ b/tests/ast/data/DataFrame.join.right_outer.predicate.test @@ -218,6 +218,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.join_table_function.test b/tests/ast/data/DataFrame.join_table_function.test index 2ca099bc60..c847ebb7be 100644 --- a/tests/ast/data/DataFrame.join_table_function.test +++ b/tests/ast/data/DataFrame.join_table_function.test @@ -263,6 +263,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.lateral_join.test b/tests/ast/data/DataFrame.lateral_join.test index a1d6ab94f9..81ccdf14c1 100644 --- a/tests/ast/data/DataFrame.lateral_join.test +++ b/tests/ast/data/DataFrame.lateral_join.test @@ -666,6 +666,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.natural_join.test b/tests/ast/data/DataFrame.natural_join.test index 9cc8844897..f890fbb451 100644 --- a/tests/ast/data/DataFrame.natural_join.test +++ b/tests/ast/data/DataFrame.natural_join.test @@ -169,6 +169,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.pivot.test b/tests/ast/data/DataFrame.pivot.test index b642850a0f..42d3e0b6e5 100644 --- a/tests/ast/data/DataFrame.pivot.test +++ b/tests/ast/data/DataFrame.pivot.test @@ -756,6 +756,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.select_expr.test b/tests/ast/data/DataFrame.select_expr.test index 40f0c4cb38..23fb964e1b 100644 --- a/tests/ast/data/DataFrame.select_expr.test +++ b/tests/ast/data/DataFrame.select_expr.test @@ -359,6 +359,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.stat.test b/tests/ast/data/DataFrame.stat.test index cda4b49864..a4f3f85cfa 100644 --- a/tests/ast/data/DataFrame.stat.test +++ b/tests/ast/data/DataFrame.stat.test @@ -1647,6 +1647,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.to_df.test b/tests/ast/data/DataFrame.to_df.test index 2f99e5cee8..865bdbd2e4 100644 --- a/tests/ast/data/DataFrame.to_df.test +++ b/tests/ast/data/DataFrame.to_df.test @@ -172,6 +172,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.to_local_iterator.test b/tests/ast/data/DataFrame.to_local_iterator.test index 37fffb9139..c6d870eaa7 100644 --- a/tests/ast/data/DataFrame.to_local_iterator.test +++ b/tests/ast/data/DataFrame.to_local_iterator.test @@ -539,6 +539,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.to_pandas.test b/tests/ast/data/DataFrame.to_pandas.test index ef2309c197..c73ffd70fb 100644 --- a/tests/ast/data/DataFrame.to_pandas.test +++ b/tests/ast/data/DataFrame.to_pandas.test @@ -298,6 +298,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.to_pandas_batch.test b/tests/ast/data/DataFrame.to_pandas_batch.test index 2f7c86ef42..6e795eb939 100644 --- a/tests/ast/data/DataFrame.to_pandas_batch.test +++ b/tests/ast/data/DataFrame.to_pandas_batch.test @@ -298,6 +298,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.unpivot.test b/tests/ast/data/DataFrame.unpivot.test index ad8cf5c0c6..4ce607a89d 100644 --- a/tests/ast/data/DataFrame.unpivot.test +++ b/tests/ast/data/DataFrame.unpivot.test @@ -237,6 +237,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataFrame.write.test b/tests/ast/data/DataFrame.write.test index d1135b0075..fd0541069c 100644 --- a/tests/ast/data/DataFrame.write.test +++ b/tests/ast/data/DataFrame.write.test @@ -6237,6 +6237,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.cube.test b/tests/ast/data/Dataframe.cube.test index f2230e6fcd..c706996159 100644 --- a/tests/ast/data/Dataframe.cube.test +++ b/tests/ast/data/Dataframe.cube.test @@ -520,6 +520,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.distinct.test b/tests/ast/data/Dataframe.distinct.test index 7d2cce9052..25cfe1615a 100644 --- a/tests/ast/data/Dataframe.distinct.test +++ b/tests/ast/data/Dataframe.distinct.test @@ -89,6 +89,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.drop_duplicates.test b/tests/ast/data/Dataframe.drop_duplicates.test index 610e6df02f..48f7d4d7e6 100644 --- a/tests/ast/data/Dataframe.drop_duplicates.test +++ b/tests/ast/data/Dataframe.drop_duplicates.test @@ -278,6 +278,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.drop_duplicates_snow1874622.test b/tests/ast/data/Dataframe.drop_duplicates_snow1874622.test index 0cbf8fbdab..e247ba1e78 100644 --- a/tests/ast/data/Dataframe.drop_duplicates_snow1874622.test +++ b/tests/ast/data/Dataframe.drop_duplicates_snow1874622.test @@ -525,6 +525,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.filter.test b/tests/ast/data/Dataframe.filter.test index 87fe31a0e4..ad8cb56d76 100644 --- a/tests/ast/data/Dataframe.filter.test +++ b/tests/ast/data/Dataframe.filter.test @@ -382,6 +382,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.getitem.test b/tests/ast/data/Dataframe.getitem.test index e9656e7e2c..ea6cf1e4b6 100644 --- a/tests/ast/data/Dataframe.getitem.test +++ b/tests/ast/data/Dataframe.getitem.test @@ -273,6 +273,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.group_by.test b/tests/ast/data/Dataframe.group_by.test index b6c586f465..120e4068b0 100644 --- a/tests/ast/data/Dataframe.group_by.test +++ b/tests/ast/data/Dataframe.group_by.test @@ -520,6 +520,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.group_by_grouping_sets.test b/tests/ast/data/Dataframe.group_by_grouping_sets.test index e2b0326970..2ef726b34f 100644 --- a/tests/ast/data/Dataframe.group_by_grouping_sets.test +++ b/tests/ast/data/Dataframe.group_by_grouping_sets.test @@ -1102,6 +1102,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.join.asof.test b/tests/ast/data/Dataframe.join.asof.test index 2f76035774..d76c013279 100644 --- a/tests/ast/data/Dataframe.join.asof.test +++ b/tests/ast/data/Dataframe.join.asof.test @@ -734,6 +734,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.join.prefix.test b/tests/ast/data/Dataframe.join.prefix.test index d92e77f3e1..e5deb3e006 100644 --- a/tests/ast/data/Dataframe.join.prefix.test +++ b/tests/ast/data/Dataframe.join.prefix.test @@ -817,6 +817,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.rollup.test b/tests/ast/data/Dataframe.rollup.test index 990934c07b..2e4a969a26 100644 --- a/tests/ast/data/Dataframe.rollup.test +++ b/tests/ast/data/Dataframe.rollup.test @@ -520,6 +520,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.show.test b/tests/ast/data/Dataframe.show.test index be64390875..e47882f56f 100644 --- a/tests/ast/data/Dataframe.show.test +++ b/tests/ast/data/Dataframe.show.test @@ -1132,6 +1132,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.to_snowpark_pandas.test.DISABLED b/tests/ast/data/Dataframe.to_snowpark_pandas.test.DISABLED index 1c5388998a..c98fb2bdd1 100644 --- a/tests/ast/data/Dataframe.to_snowpark_pandas.test.DISABLED +++ b/tests/ast/data/Dataframe.to_snowpark_pandas.test.DISABLED @@ -183,6 +183,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Dataframe.with_col_fns.test b/tests/ast/data/Dataframe.with_col_fns.test index c1cf69150e..6858f8ba40 100644 --- a/tests/ast/data/Dataframe.with_col_fns.test +++ b/tests/ast/data/Dataframe.with_col_fns.test @@ -928,6 +928,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/DataframeNaFunctions.test b/tests/ast/data/DataframeNaFunctions.test index c7d0fd8730..523f071311 100644 --- a/tests/ast/data/DataframeNaFunctions.test +++ b/tests/ast/data/DataframeNaFunctions.test @@ -882,6 +882,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/RelationalGroupedDataFrame.agg.test b/tests/ast/data/RelationalGroupedDataFrame.agg.test index e06a91ec9a..997251c2bc 100644 --- a/tests/ast/data/RelationalGroupedDataFrame.agg.test +++ b/tests/ast/data/RelationalGroupedDataFrame.agg.test @@ -1144,6 +1144,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/RelationalGroupedDataFrame.ai_agg.test b/tests/ast/data/RelationalGroupedDataFrame.ai_agg.test index 5f82d421b7..fde894d499 100644 --- a/tests/ast/data/RelationalGroupedDataFrame.ai_agg.test +++ b/tests/ast/data/RelationalGroupedDataFrame.ai_agg.test @@ -287,6 +287,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/RelationalGroupedDataFrame.test b/tests/ast/data/RelationalGroupedDataFrame.test index 71fbc4184e..4b840f6e0f 100644 --- a/tests/ast/data/RelationalGroupedDataFrame.test +++ b/tests/ast/data/RelationalGroupedDataFrame.test @@ -3621,6 +3621,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Session.call.test b/tests/ast/data/Session.call.test index 4dfdb376c4..68e2d08e75 100644 --- a/tests/ast/data/Session.call.test +++ b/tests/ast/data/Session.call.test @@ -360,6 +360,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Session.create_dataframe.test b/tests/ast/data/Session.create_dataframe.test index 1e5b849868..51377ae849 100644 --- a/tests/ast/data/Session.create_dataframe.test +++ b/tests/ast/data/Session.create_dataframe.test @@ -1272,6 +1272,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Session.create_dataframe_from_pandas.test b/tests/ast/data/Session.create_dataframe_from_pandas.test index 22e31574af..77fa797a6f 100644 --- a/tests/ast/data/Session.create_dataframe_from_pandas.test +++ b/tests/ast/data/Session.create_dataframe_from_pandas.test @@ -171,6 +171,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Session.flatten.test b/tests/ast/data/Session.flatten.test index c395d4b319..6d2e253c18 100644 --- a/tests/ast/data/Session.flatten.test +++ b/tests/ast/data/Session.flatten.test @@ -174,6 +174,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Session.table_function.test b/tests/ast/data/Session.table_function.test index ef22bd9b23..65d393a3bb 100644 --- a/tests/ast/data/Session.table_function.test +++ b/tests/ast/data/Session.table_function.test @@ -990,6 +990,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Table.delete.test b/tests/ast/data/Table.delete.test index e2f246f593..1fa695ecc0 100644 --- a/tests/ast/data/Table.delete.test +++ b/tests/ast/data/Table.delete.test @@ -545,6 +545,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Table.drop_table.test b/tests/ast/data/Table.drop_table.test index cfadde11c0..2477acffcf 100644 --- a/tests/ast/data/Table.drop_table.test +++ b/tests/ast/data/Table.drop_table.test @@ -93,6 +93,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Table.init.test b/tests/ast/data/Table.init.test index 54fa8ba68e..3b50d0db64 100644 --- a/tests/ast/data/Table.init.test +++ b/tests/ast/data/Table.init.test @@ -423,6 +423,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Table.merge.test b/tests/ast/data/Table.merge.test index 7260506af6..028ebdd33e 100644 --- a/tests/ast/data/Table.merge.test +++ b/tests/ast/data/Table.merge.test @@ -2448,6 +2448,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Table.sample.test b/tests/ast/data/Table.sample.test index 539752530a..191c7c8b9c 100644 --- a/tests/ast/data/Table.sample.test +++ b/tests/ast/data/Table.sample.test @@ -168,6 +168,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/Table.update.test b/tests/ast/data/Table.update.test index 9613d2db5d..42d46c184e 100644 --- a/tests/ast/data/Table.update.test +++ b/tests/ast/data/Table.update.test @@ -1030,6 +1030,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/case_when.test b/tests/ast/data/case_when.test index 6372082cc1..d1df9368b8 100644 --- a/tests/ast/data/case_when.test +++ b/tests/ast/data/case_when.test @@ -1913,6 +1913,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_alias.test b/tests/ast/data/col_alias.test index e3e81ac91a..c61c0ad7aa 100644 --- a/tests/ast/data/col_alias.test +++ b/tests/ast/data/col_alias.test @@ -409,6 +409,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_asc.test b/tests/ast/data/col_asc.test index d8381bef23..8c4bb5237f 100644 --- a/tests/ast/data/col_asc.test +++ b/tests/ast/data/col_asc.test @@ -302,6 +302,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_between.test b/tests/ast/data/col_between.test index cb5d878008..60d0d3d8b9 100644 --- a/tests/ast/data/col_between.test +++ b/tests/ast/data/col_between.test @@ -184,6 +184,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_binops.test b/tests/ast/data/col_binops.test index 6c2c59483a..21039fb1ba 100644 --- a/tests/ast/data/col_binops.test +++ b/tests/ast/data/col_binops.test @@ -1615,6 +1615,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_bitops.test b/tests/ast/data/col_bitops.test index a161e4234f..1aabd0b854 100644 --- a/tests/ast/data/col_bitops.test +++ b/tests/ast/data/col_bitops.test @@ -394,6 +394,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_cast.test b/tests/ast/data/col_cast.test index a5f5e16517..17e0702dd0 100644 --- a/tests/ast/data/col_cast.test +++ b/tests/ast/data/col_cast.test @@ -2854,6 +2854,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_cast_coll.test b/tests/ast/data/col_cast_coll.test index f99e34c96f..f7634348e7 100644 --- a/tests/ast/data/col_cast_coll.test +++ b/tests/ast/data/col_cast_coll.test @@ -806,6 +806,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_desc.test b/tests/ast/data/col_desc.test index 3754612245..be023a0465 100644 --- a/tests/ast/data/col_desc.test +++ b/tests/ast/data/col_desc.test @@ -300,6 +300,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_getitem.test b/tests/ast/data/col_getitem.test index 025028a7e6..d56e1b489b 100644 --- a/tests/ast/data/col_getitem.test +++ b/tests/ast/data/col_getitem.test @@ -185,6 +185,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_in_.test b/tests/ast/data/col_in_.test index 3f8651a2a8..08cb8e56e8 100644 --- a/tests/ast/data/col_in_.test +++ b/tests/ast/data/col_in_.test @@ -358,6 +358,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_lit.test b/tests/ast/data/col_lit.test index fe1006f1ee..4e13a9039f 100644 --- a/tests/ast/data/col_lit.test +++ b/tests/ast/data/col_lit.test @@ -528,6 +528,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_literal.test b/tests/ast/data/col_literal.test index 029b4222b0..ff75b0d606 100644 --- a/tests/ast/data/col_literal.test +++ b/tests/ast/data/col_literal.test @@ -3709,6 +3709,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_null_nan.test b/tests/ast/data/col_null_nan.test index e0be447193..63362674bd 100644 --- a/tests/ast/data/col_null_nan.test +++ b/tests/ast/data/col_null_nan.test @@ -403,6 +403,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_rbinops.test b/tests/ast/data/col_rbinops.test index 55a719a9be..3ccd40fd40 100644 --- a/tests/ast/data/col_rbinops.test +++ b/tests/ast/data/col_rbinops.test @@ -773,6 +773,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_star.test b/tests/ast/data/col_star.test index f94a426a59..4c671d6d25 100644 --- a/tests/ast/data/col_star.test +++ b/tests/ast/data/col_star.test @@ -127,6 +127,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_string.test b/tests/ast/data/col_string.test index 839e799c5e..a9a5436174 100644 --- a/tests/ast/data/col_string.test +++ b/tests/ast/data/col_string.test @@ -740,6 +740,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_try_cast.test b/tests/ast/data/col_try_cast.test index a0859a8c4d..8c32889bf7 100644 --- a/tests/ast/data/col_try_cast.test +++ b/tests/ast/data/col_try_cast.test @@ -2514,6 +2514,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_udf.test b/tests/ast/data/col_udf.test index a3f4013463..3f949de704 100644 --- a/tests/ast/data/col_udf.test +++ b/tests/ast/data/col_udf.test @@ -1487,6 +1487,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_unary_ops.test b/tests/ast/data/col_unary_ops.test index 746bea68ea..984850f190 100644 --- a/tests/ast/data/col_unary_ops.test +++ b/tests/ast/data/col_unary_ops.test @@ -215,6 +215,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/col_within_group.test b/tests/ast/data/col_within_group.test index e7b042d407..d9b7b3a553 100644 --- a/tests/ast/data/col_within_group.test +++ b/tests/ast/data/col_within_group.test @@ -431,6 +431,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_alias.test b/tests/ast/data/df_alias.test index 52df91a06f..2d5910f15c 100644 --- a/tests/ast/data/df_alias.test +++ b/tests/ast/data/df_alias.test @@ -90,6 +90,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_analytics_functions.test b/tests/ast/data/df_analytics_functions.test index 2e22398ffd..f980508b48 100644 --- a/tests/ast/data/df_analytics_functions.test +++ b/tests/ast/data/df_analytics_functions.test @@ -950,6 +950,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_col.test b/tests/ast/data/df_col.test index 10943c5957..57ac3cc15c 100644 --- a/tests/ast/data/df_col.test +++ b/tests/ast/data/df_col.test @@ -144,6 +144,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_copy.test b/tests/ast/data/df_copy.test index b98df007d9..0ebd830068 100644 --- a/tests/ast/data/df_copy.test +++ b/tests/ast/data/df_copy.test @@ -778,6 +778,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_drop.test b/tests/ast/data/df_drop.test index f277ee2b95..f57c3a5be3 100644 --- a/tests/ast/data/df_drop.test +++ b/tests/ast/data/df_drop.test @@ -127,6 +127,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_except.test b/tests/ast/data/df_except.test index 9f68548be8..5af3a84d5f 100644 --- a/tests/ast/data/df_except.test +++ b/tests/ast/data/df_except.test @@ -128,6 +128,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_first.test b/tests/ast/data/df_first.test index b8a69150f7..c55d7e8b24 100644 --- a/tests/ast/data/df_first.test +++ b/tests/ast/data/df_first.test @@ -383,6 +383,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_intersect.test b/tests/ast/data/df_intersect.test index 30286b5e6a..e7eaffb72c 100644 --- a/tests/ast/data/df_intersect.test +++ b/tests/ast/data/df_intersect.test @@ -129,6 +129,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_limit.test b/tests/ast/data/df_limit.test index 140b20377e..0f0cb665a9 100644 --- a/tests/ast/data/df_limit.test +++ b/tests/ast/data/df_limit.test @@ -342,6 +342,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_map.test b/tests/ast/data/df_map.test index dd7480c8bb..9568709f3a 100644 --- a/tests/ast/data/df_map.test +++ b/tests/ast/data/df_map.test @@ -2142,6 +2142,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_random_split.test b/tests/ast/data/df_random_split.test index f0fce1fe7d..c8bb5e97a8 100644 --- a/tests/ast/data/df_random_split.test +++ b/tests/ast/data/df_random_split.test @@ -710,6 +710,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_sample.test b/tests/ast/data/df_sample.test index 4f9fca24f8..036fb156ba 100644 --- a/tests/ast/data/df_sample.test +++ b/tests/ast/data/df_sample.test @@ -124,6 +124,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_sort.test b/tests/ast/data/df_sort.test index 72b58f725e..de162e4c00 100644 --- a/tests/ast/data/df_sort.test +++ b/tests/ast/data/df_sort.test @@ -833,6 +833,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/df_union.test b/tests/ast/data/df_union.test index 5d642c14df..0302a0880c 100644 --- a/tests/ast/data/df_union.test +++ b/tests/ast/data/df_union.test @@ -375,6 +375,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/functions.table_functions.test b/tests/ast/data/functions.table_functions.test index 86f2339f34..01dd7fd8ac 100644 --- a/tests/ast/data/functions.table_functions.test +++ b/tests/ast/data/functions.table_functions.test @@ -2212,6 +2212,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/functions1.test b/tests/ast/data/functions1.test index cf4cc172f0..f41f20b02e 100644 --- a/tests/ast/data/functions1.test +++ b/tests/ast/data/functions1.test @@ -18800,6 +18800,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/functions2.test b/tests/ast/data/functions2.test index 8de51cde05..f415d58bc1 100644 --- a/tests/ast/data/functions2.test +++ b/tests/ast/data/functions2.test @@ -29612,6 +29612,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/interval.test b/tests/ast/data/interval.test index c0aa8d0811..ed5334aad0 100644 --- a/tests/ast/data/interval.test +++ b/tests/ast/data/interval.test @@ -1069,6 +1069,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/select.test b/tests/ast/data/select.test index 54ae3aedca..898237f56f 100644 --- a/tests/ast/data/select.test +++ b/tests/ast/data/select.test @@ -139,6 +139,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session.read.test b/tests/ast/data/session.read.test index a8f99a1f5d..f75d333c4b 100644 --- a/tests/ast/data/session.read.test +++ b/tests/ast/data/session.read.test @@ -1772,6 +1772,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session.sql.test b/tests/ast/data/session.sql.test index 9a11335af4..c215ffa78f 100644 --- a/tests/ast/data/session.sql.test +++ b/tests/ast/data/session.sql.test @@ -149,6 +149,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_directory.test b/tests/ast/data/session_directory.test index 626b5a45f1..066eb73604 100644 --- a/tests/ast/data/session_directory.test +++ b/tests/ast/data/session_directory.test @@ -101,6 +101,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_generator.test b/tests/ast/data/session_generator.test index de12fec065..5982344368 100644 --- a/tests/ast/data/session_generator.test +++ b/tests/ast/data/session_generator.test @@ -362,6 +362,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_range.test b/tests/ast/data/session_range.test index 83381e8aa5..a294d0ae03 100644 --- a/tests/ast/data/session_range.test +++ b/tests/ast/data/session_range.test @@ -172,6 +172,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_dq_abs_l.test b/tests/ast/data/session_table_dq_abs_l.test index 17a5990438..46a43b0006 100644 --- a/tests/ast/data/session_table_dq_abs_l.test +++ b/tests/ast/data/session_table_dq_abs_l.test @@ -105,6 +105,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_dq_abs_s.test b/tests/ast/data/session_table_dq_abs_s.test index aefbd172c1..42f489dedd 100644 --- a/tests/ast/data/session_table_dq_abs_s.test +++ b/tests/ast/data/session_table_dq_abs_s.test @@ -103,6 +103,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_dq_rs_l.test b/tests/ast/data/session_table_dq_rs_l.test index 4da8f9fa2a..3f8951292a 100644 --- a/tests/ast/data/session_table_dq_rs_l.test +++ b/tests/ast/data/session_table_dq_rs_l.test @@ -104,6 +104,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_dq_rs_s.test b/tests/ast/data/session_table_dq_rs_s.test index ac477be967..3c3dcaa71d 100644 --- a/tests/ast/data/session_table_dq_rs_s.test +++ b/tests/ast/data/session_table_dq_rs_s.test @@ -103,6 +103,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_dq_rt_l.test b/tests/ast/data/session_table_dq_rt_l.test index f4f86c62af..1d29ec326a 100644 --- a/tests/ast/data/session_table_dq_rt_l.test +++ b/tests/ast/data/session_table_dq_rt_l.test @@ -103,6 +103,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_dq_rt_s.test b/tests/ast/data/session_table_dq_rt_s.test index 52a32b6a10..0279b2c64e 100644 --- a/tests/ast/data/session_table_dq_rt_s.test +++ b/tests/ast/data/session_table_dq_rt_s.test @@ -103,6 +103,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_temp_table_cleanup.test b/tests/ast/data/session_table_temp_table_cleanup.test index 4cd527b92f..223fd37c9e 100644 --- a/tests/ast/data/session_table_temp_table_cleanup.test +++ b/tests/ast/data/session_table_temp_table_cleanup.test @@ -170,6 +170,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_uq_abs_l.test b/tests/ast/data/session_table_uq_abs_l.test index 7da6d54685..d6d59411bf 100644 --- a/tests/ast/data/session_table_uq_abs_l.test +++ b/tests/ast/data/session_table_uq_abs_l.test @@ -105,6 +105,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_uq_abs_s.test b/tests/ast/data/session_table_uq_abs_s.test index 05ff448424..f6817f3450 100644 --- a/tests/ast/data/session_table_uq_abs_s.test +++ b/tests/ast/data/session_table_uq_abs_s.test @@ -103,6 +103,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_uq_rs_l.test b/tests/ast/data/session_table_uq_rs_l.test index 97e2a776ee..dd5e3de6cd 100644 --- a/tests/ast/data/session_table_uq_rs_l.test +++ b/tests/ast/data/session_table_uq_rs_l.test @@ -104,6 +104,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_uq_rs_s.test b/tests/ast/data/session_table_uq_rs_s.test index 627e847c31..95b4775040 100644 --- a/tests/ast/data/session_table_uq_rs_s.test +++ b/tests/ast/data/session_table_uq_rs_s.test @@ -103,6 +103,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_uq_rt_l.test b/tests/ast/data/session_table_uq_rt_l.test index 351ea25576..b5428e38ac 100644 --- a/tests/ast/data/session_table_uq_rt_l.test +++ b/tests/ast/data/session_table_uq_rt_l.test @@ -103,6 +103,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_table_uq_rt_s.test b/tests/ast/data/session_table_uq_rt_s.test index 2a834c5469..7b4b0dfbe0 100644 --- a/tests/ast/data/session_table_uq_rt_s.test +++ b/tests/ast/data/session_table_uq_rt_s.test @@ -388,6 +388,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_transaction.test b/tests/ast/data/session_transaction.test index 51ca313076..0ab61baccc 100644 --- a/tests/ast/data/session_transaction.test +++ b/tests/ast/data/session_transaction.test @@ -121,6 +121,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/session_write_pandas.test b/tests/ast/data/session_write_pandas.test index a1eb9554f5..fb0300efcc 100644 --- a/tests/ast/data/session_write_pandas.test +++ b/tests/ast/data/session_write_pandas.test @@ -149,6 +149,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/shadowed_local_name.test b/tests/ast/data/shadowed_local_name.test index 09929b5854..b2a633acf0 100644 --- a/tests/ast/data/shadowed_local_name.test +++ b/tests/ast/data/shadowed_local_name.test @@ -205,6 +205,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/smoke1.test b/tests/ast/data/smoke1.test index 2523fe5ab9..59e0cc84eb 100644 --- a/tests/ast/data/smoke1.test +++ b/tests/ast/data/smoke1.test @@ -2188,6 +2188,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/smoke2.test b/tests/ast/data/smoke2.test index 2ca01b63b1..1adaf6c579 100644 --- a/tests/ast/data/smoke2.test +++ b/tests/ast/data/smoke2.test @@ -2352,6 +2352,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/sproc.test b/tests/ast/data/sproc.test index 1bd73b4190..bd4dd8ea11 100644 --- a/tests/ast/data/sproc.test +++ b/tests/ast/data/sproc.test @@ -2828,6 +2828,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/udaf.test b/tests/ast/data/udaf.test index fb9961a33a..5bf272aa84 100644 --- a/tests/ast/data/udaf.test +++ b/tests/ast/data/udaf.test @@ -1277,6 +1277,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/udtf.test b/tests/ast/data/udtf.test index 7fff7db786..40e4b31067 100644 --- a/tests/ast/data/udtf.test +++ b/tests/ast/data/udtf.test @@ -3580,6 +3580,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/ast/data/windows.test b/tests/ast/data/windows.test index 7005d31c7f..4b6b7b0afc 100644 --- a/tests/ast/data/windows.test +++ b/tests/ast/data/windows.test @@ -2091,6 +2091,6 @@ client_language { } client_version { major: 1 - minor: 45 + minor: 46 } id: "\003U\"\366q\366P\346\260\261?\234\303\254\316\353" diff --git a/tests/integ/test_packaging.py b/tests/integ/test_packaging.py index 02998f4a3a..f73bcb09fa 100644 --- a/tests/integ/test_packaging.py +++ b/tests/integ/test_packaging.py @@ -271,8 +271,8 @@ def extract_major_minor_patch(version_string): resolved_packages = session._resolve_packages( [numpy, pandas, dateutil], - _ANACONDA_SHARED_REPOSITORY, - {}, + artifact_repository=_ANACONDA_SHARED_REPOSITORY, + existing_packages_dict={}, validate_package=False, ) # resolved_packages is a list of strings like @@ -1204,17 +1204,10 @@ def test_replicate_local_environment(session): "force_push": True, } - assert not any( - [ - package.startswith("cloudpickle") - for package in session._artifact_repository_packages[ - _ANACONDA_SHARED_REPOSITORY - ] - ] - ) + assert not any([package.startswith("cloudpickle") for package in session._packages]) def naive_add_packages(self, packages): - self._artifact_repository_packages[_ANACONDA_SHARED_REPOSITORY] = packages + self._packages = packages with patch.object(session, "_is_anaconda_terms_acknowledged", lambda: True): with patch.object(Session, "add_packages", new=naive_add_packages): @@ -1228,22 +1221,10 @@ def naive_add_packages(self, packages): }, ) - assert any( - [ - package.startswith("cloudpickle==") - for package in session._artifact_repository_packages[ - _ANACONDA_SHARED_REPOSITORY - ] - ] - ) + assert any([package.startswith("cloudpickle==") for package in session._packages]) for default_package in DEFAULT_PACKAGES: assert not any( - [ - package.startswith(default_package) - for package in session._artifact_repository_packages[ - _ANACONDA_SHARED_REPOSITORY - ] - ] + [package.startswith(default_package) for package in session._packages] ) session.clear_packages() @@ -1262,29 +1243,12 @@ def naive_add_packages(self, packages): ignore_packages=ignored_packages, relax=True ) - assert any( - [ - package == "cloudpickle" - for package in session._artifact_repository_packages[ - _ANACONDA_SHARED_REPOSITORY - ] - ] - ) + assert any([package == "cloudpickle" for package in session._packages]) for default_package in DEFAULT_PACKAGES: assert not any( - [ - package.startswith(default_package) - for package in session._artifact_repository_packages[ - _ANACONDA_SHARED_REPOSITORY - ] - ] + [package.startswith(default_package) for package in session._packages] ) for ignored_package in ignored_packages: assert not any( - [ - package.startswith(ignored_package) - for package in session._artifact_repository_packages[ - _ANACONDA_SHARED_REPOSITORY - ] - ] + [package.startswith(ignored_package) for package in session._packages] ) diff --git a/tests/integ/test_udf.py b/tests/integ/test_udf.py index 0499a01a3d..d9029efd23 100644 --- a/tests/integ/test_udf.py +++ b/tests/integ/test_udf.py @@ -3108,18 +3108,17 @@ def test_urllib() -> str: ) def test_use_default_artifact_repository(db_parameters): with Session.builder.configs(db_parameters).create() as session: + temp_database = Utils.random_temp_database() temp_schema = Utils.random_temp_schema() - session.sql(f"create schema {temp_schema}").collect() - session.sql(f"use schema {temp_schema}").collect() + session.sql(f"create database {temp_database}").collect() + session.sql(f"use database {temp_database}").collect() session.sql( "ALTER SESSION SET ENABLE_DEFAULT_PYTHON_ARTIFACT_REPOSITORY = true" ).collect() session.sql( - "ALTER schema set DEFAULT_PYTHON_ARTIFACT_REPOSITORY = testdb_snowpark_python.testschema_snowpark_python.SNOWPARK_PYTHON_TEST_REPOSITORY" + "ALTER database set DEFAULT_PYTHON_ARTIFACT_REPOSITORY = snowflake.snowpark.anaconda_shared_repository" ).collect() - session.add_packages("art", "cloudpickle") - def test_art() -> str: import art # art is not available in the conda channel, but is in pypi @@ -3128,6 +3127,25 @@ def test_art() -> str: temp_func_name = Utils.random_name_for_temp_object(TempObjectType.FUNCTION) + # should not work in the database where the default is anaconda + with pytest.raises( + Exception, + match="Cannot add package art because it is not available in Snowflake", + ): + udf( + session=session, + func=test_art, + name=temp_func_name, + packages=["art", "cloudpickle"], + ) + + session.sql(f"create schema {temp_schema}").collect() + session.use_schema(temp_schema) + session.sql( + "ALTER schema set DEFAULT_PYTHON_ARTIFACT_REPOSITORY = testdb_snowpark_python.testschema_snowpark_python.SNOWPARK_PYTHON_TEST_REPOSITORY" + ).collect() + session.add_packages("art", "cloudpickle") + try: # Test function registration udf( @@ -3142,4 +3160,4 @@ def test_art() -> str: finally: session._run_query(f"drop function if exists {temp_func_name}(int)") - session.sql(f"drop schema {temp_schema}").collect() + session.sql(f"drop database {temp_database}").collect() diff --git a/tests/unit/test_session.py b/tests/unit/test_session.py index 7a64f052d4..69cf7b6ac2 100644 --- a/tests/unit/test_session.py +++ b/tests/unit/test_session.py @@ -213,8 +213,8 @@ def mock_get_information_schema_packages(table_name: str, _emit_ast: bool = True session._resolve_packages( ["random_package_name"], - _ANACONDA_SHARED_REPOSITORY, - {}, + artifact_repository=_ANACONDA_SHARED_REPOSITORY, + existing_packages_dict={}, validate_package=True, include_pandas=False, ) @@ -248,8 +248,8 @@ def run_query(sql: str): ): session._resolve_packages( ["random_package_name"], - _ANACONDA_SHARED_REPOSITORY, - {}, + artifact_repository=_ANACONDA_SHARED_REPOSITORY, + existing_packages_dict={}, validate_package=True, include_pandas=False, ) @@ -273,8 +273,8 @@ def mock_get_information_schema_packages(table_name: str, _emit_ast: bool = True resolved_packages = session._resolve_packages( ["random_package_name"], - _ANACONDA_SHARED_REPOSITORY, - existing_packages, + artifact_repository=_ANACONDA_SHARED_REPOSITORY, + existing_packages_dict=existing_packages, validate_package=True, include_pandas=False, ) @@ -305,8 +305,8 @@ def mock_get_information_schema_packages(table_name: str, _emit_ast: bool = True ): session._resolve_packages( ["snowflake-snowpark-python"], - _ANACONDA_SHARED_REPOSITORY, - {}, + artifact_repository=_ANACONDA_SHARED_REPOSITORY, + existing_packages_dict={}, validate_package=True, include_pandas=False, _suppress_local_package_warnings=True, @@ -333,21 +333,41 @@ def assert_packages(packages): packages = session._resolve_packages( ["snowflake-snowpark-python==1.0.0", "cloudpickle==1.0.0"], - "snowflake.snowpark.pypi_shared_repository", - existing_packages, + artifact_repository="snowflake.snowpark.pypi_shared_repository", + existing_packages_dict=existing_packages, ) assert_packages(packages) packages = session._resolve_packages( [], - "snowflake.snowpark.pypi_shared_repository", - existing_packages, + artifact_repository="snowflake.snowpark.pypi_shared_repository", + existing_packages_dict=existing_packages, ) assert_packages(packages) +def test_resolve_packages_optional_artifact_repository(mock_server_connection): + session = Session(mock_server_connection) + session._get_default_artifact_repository = MagicMock( + return_value="snowflake.snowpark.pypi_shared_repository" + ) + session._artifact_repository_packages = { + "snowflake.snowpark.pypi_shared_repository": { + "numpy": "numpy==1.0.0", + } + } + result = session._resolve_packages( + ["snowflake-snowpark-python==1.0.0", "cloudpickle==1.0.0"], + ) + assert sorted(result) == [ + "cloudpickle==1.0.0", + "numpy==1.0.0", + "snowflake-snowpark-python==1.0.0", + ] + + @pytest.mark.skipif(not is_pandas_available, reason="requires pandas for write_pandas") def test_write_pandas_wrong_table_type(mock_server_connection): session = Session(mock_server_connection) diff --git a/tests/unit/test_stored_procedure.py b/tests/unit/test_stored_procedure.py index 2e6aff999a..eb9917f682 100644 --- a/tests/unit/test_stored_procedure.py +++ b/tests/unit/test_stored_procedure.py @@ -44,6 +44,7 @@ def test_stored_procedure_execute_as(execute_as): fake_session._analyzer = Analyzer(fake_session) fake_session._runtime_version_from_requirement = None fake_session._artifact_repository_packages = defaultdict(dict) + fake_session._packages = {} def return1(_): return 1 @@ -92,6 +93,7 @@ def test_do_register_sp_negative(cleanup_registration_patch): fake_session._run_query = mock.Mock(side_effect=ProgrammingError()) fake_session.sproc = StoredProcedureRegistration(fake_session) fake_session._artifact_repository_packages = defaultdict(dict) + fake_session._packages = {} with pytest.raises(SnowparkSQLException) as ex_info: sproc(lambda: 1, session=fake_session, return_type=IntegerType(), packages=[]) assert ex_info.value.error_code == "1304" diff --git a/tests/unit/test_udaf.py b/tests/unit/test_udaf.py index 803f8dc886..464be0c856 100644 --- a/tests/unit/test_udaf.py +++ b/tests/unit/test_udaf.py @@ -59,6 +59,7 @@ def test_do_register_udaf_negative(cleanup_registration_patch): fake_session._runtime_version_from_requirement = None fake_session.udaf = UDAFRegistration(fake_session) fake_session._artifact_repository_packages = defaultdict(dict) + fake_session._packages = {} with pytest.raises(SnowparkSQLException) as ex_info: @udaf(session=fake_session) diff --git a/tests/unit/test_udf.py b/tests/unit/test_udf.py index d20c53d048..b3c70bfc70 100644 --- a/tests/unit/test_udf.py +++ b/tests/unit/test_udf.py @@ -34,6 +34,7 @@ def test_do_register_sp_negative(cleanup_registration_patch): fake_session._run_query = mock.Mock(side_effect=ProgrammingError()) fake_session.udf = UDFRegistration(fake_session) fake_session._artifact_repository_packages = defaultdict(dict) + fake_session._packages = {} with pytest.raises(SnowparkSQLException) as ex_info: udf(lambda: 1, session=fake_session, return_type=IntegerType(), packages=[]) assert ex_info.value.error_code == "1304" diff --git a/tests/unit/test_udf_utils.py b/tests/unit/test_udf_utils.py index a301d8769e..32a1626eb9 100644 --- a/tests/unit/test_udf_utils.py +++ b/tests/unit/test_udf_utils.py @@ -2,7 +2,6 @@ # Copyright (c) 2012-2025 Snowflake Computing Inc. All rights reserved. # -from collections import defaultdict import logging import os import pickle @@ -254,12 +253,14 @@ def test_add_snowpark_package_to_sproc_packages_does_not_replace_package(): def test_add_snowpark_package_to_sproc_packages_to_session(): fake_session = mock.create_autospec(Session) - fake_session._artifact_repository_packages = defaultdict(dict) - fake_session._artifact_repository_packages[_ANACONDA_SHARED_REPOSITORY] = { + fake_session._packages = { "random_package_one": "random_package_one", "random_package_two": "random_package_two", } fake_session._package_lock = threading.RLock() + fake_session._get_packages_by_artifact_repository.side_effect = ( + lambda a: Session._get_packages_by_artifact_repository(fake_session, a) + ) result = add_snowpark_package_to_sproc_packages( session=fake_session, packages=None, @@ -272,7 +273,7 @@ def test_add_snowpark_package_to_sproc_packages_to_session(): assert len(result) == 3 assert final_name in result - fake_session._artifact_repository_packages[_ANACONDA_SHARED_REPOSITORY][ + fake_session._packages[ "snowflake-snowpark-python" ] = "snowflake-snowpark-python==1.12.0" result = add_snowpark_package_to_sproc_packages( diff --git a/tests/unit/test_udtf.py b/tests/unit/test_udtf.py index ba986364ed..e6f3585641 100644 --- a/tests/unit/test_udtf.py +++ b/tests/unit/test_udtf.py @@ -42,6 +42,7 @@ def test_do_register_sp_negative(cleanup_registration_patch): fake_session._runtime_version_from_requirement = None fake_session.udtf = UDTFRegistration(fake_session) fake_session._artifact_repository_packages = defaultdict(dict) + fake_session._packages = {} with pytest.raises(SnowparkSQLException) as ex_info: @udtf(output_schema=["num"], session=fake_session)