diff --git a/.bazelversion b/.bazelversion index 798e389..91e4a9f 100644 --- a/.bazelversion +++ b/.bazelversion @@ -1 +1 @@ -6.3.0 +6.3.2 diff --git a/CHANGELOG.md b/CHANGELOG.md index 269e9e4..27ae19f 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,6 +1,12 @@ # Release History -## 1.6.3 +## 1.6.4 + +### Bug Fixes + +- Registry: Fix an issue that leads to incident when using `ModelVersion.run` with service. + +## 1.6.3 (2024-10-07) - Model Registry (PrPr) has been removed. diff --git a/bazel/requirements/templates/meta.tpl.yaml b/bazel/requirements/templates/meta.tpl.yaml index 3655610..19405ab 100644 --- a/bazel/requirements/templates/meta.tpl.yaml +++ b/bazel/requirements/templates/meta.tpl.yaml @@ -11,7 +11,7 @@ build: requirements: build: - python - - bazel==6.3.0 + - bazel==6.3.2 run: - python>=3.8,<3.12 diff --git a/ci/conda_recipe/meta.yaml b/ci/conda_recipe/meta.yaml index c03b196..e2f8787 100644 --- a/ci/conda_recipe/meta.yaml +++ b/ci/conda_recipe/meta.yaml @@ -17,11 +17,11 @@ build: noarch: python package: name: snowflake-ml-python - version: 1.6.3 + version: 1.6.4 requirements: build: - python - - bazel==6.3.0 + - bazel==6.3.2 run: - absl-py>=0.15,<2 - aiohttp!=4.0.0a0, !=4.0.0a1 diff --git a/snowflake/ml/model/_client/sql/service.py b/snowflake/ml/model/_client/sql/service.py index 4329301..6033bff 100644 --- a/snowflake/ml/model/_client/sql/service.py +++ b/snowflake/ml/model/_client/sql/service.py @@ -3,13 +3,10 @@ import textwrap from typing import Any, Dict, List, Optional, Tuple -from packaging import version - from snowflake import snowpark from snowflake.ml._internal.utils import ( identifier, query_result_checker, - snowflake_env, sql_identifier, ) from snowflake.ml.model._client.sql import _base @@ -120,21 +117,12 @@ def invoke_function_method( args_sql_list.append(input_arg_value) args_sql = ", ".join(args_sql_list) - if snowflake_env.get_current_snowflake_version( - self._session, statement_params=statement_params - ) >= version.parse("8.39.0"): - fully_qualified_service_name = self.fully_qualified_object_name( - actual_database_name, actual_schema_name, service_name - ) - fully_qualified_function_name = f"{fully_qualified_service_name}!{method_name.identifier()}" - - else: - function_name = identifier.concat_names([service_name.identifier(), "_", method_name.identifier()]) - fully_qualified_function_name = identifier.get_schema_level_object_identifier( - actual_database_name.identifier(), - actual_schema_name.identifier(), - function_name, - ) + function_name = identifier.concat_names([service_name.identifier(), "_", method_name.identifier()]) + fully_qualified_function_name = identifier.get_schema_level_object_identifier( + actual_database_name.identifier(), + actual_schema_name.identifier(), + function_name, + ) sql = textwrap.dedent( f"""{with_sql} diff --git a/snowflake/ml/model/_client/sql/service_test.py b/snowflake/ml/model/_client/sql/service_test.py index 38806a3..385cf91 100644 --- a/snowflake/ml/model/_client/sql/service_test.py +++ b/snowflake/ml/model/_client/sql/service_test.py @@ -102,8 +102,6 @@ def test_deploy_model(self) -> None: def test_invoke_function_method(self) -> None: m_statement_params = {"test": "1"} m_df = mock_data_frame.MockDataFrame() - m_df0 = mock_data_frame.MockDataFrame(collect_result=[Row(CURRENT_VERSION="1")]) - self.m_session.add_mock_sql("SELECT CURRENT_VERSION() AS CURRENT_VERSION", m_df0) self.m_session.add_mock_sql( """SELECT *, @@ -143,9 +141,6 @@ def test_invoke_function_method(self) -> None: def test_invoke_function_method_1(self) -> None: m_statement_params = {"test": "1"} - m_df0 = mock_data_frame.MockDataFrame(collect_result=[Row(CURRENT_VERSION="1")]) - self.m_session.add_mock_sql("SELECT CURRENT_VERSION() AS CURRENT_VERSION", m_df0) - m_df = mock_data_frame.MockDataFrame() self.m_session.add_mock_sql( """SELECT *, @@ -188,8 +183,6 @@ def test_invoke_function_method_1(self) -> None: def test_invoke_function_method_2(self) -> None: m_statement_params = {"test": "1"} - m_df0 = mock_data_frame.MockDataFrame(collect_result=[Row(CURRENT_VERSION="1")]) - self.m_session.add_mock_sql("SELECT CURRENT_VERSION() AS CURRENT_VERSION", m_df0) m_df = mock_data_frame.MockDataFrame() self.m_session.add_mock_sql( """WITH SNOWPARK_ML_MODEL_INFERENCE_INPUT_ABCDEF0123 AS (query_1) diff --git a/snowflake/ml/version.bzl b/snowflake/ml/version.bzl index e1060d5..2d2065a 100644 --- a/snowflake/ml/version.bzl +++ b/snowflake/ml/version.bzl @@ -1,2 +1,2 @@ # This is parsed by regex in conda reciper meta file. Make sure not to break it. -VERSION = "1.6.3" +VERSION = "1.6.4"