snowflake-ml-python 1.7.1__py3-none-any.whl → 1.7.2__py3-none-any.whl

snowflake/ml/_internal/utils/jwt_generator.py

@@ -0,0 +1,141 @@
+import base64
+import hashlib
+import logging
+from datetime import datetime, timedelta, timezone
+from typing import Optional
+
+import jwt
+from cryptography.hazmat.primitives import serialization
+from cryptography.hazmat.primitives.asymmetric import types
+
+logger = logging.getLogger(__name__)
+
+ISSUER = "iss"
+EXPIRE_TIME = "exp"
+ISSUE_TIME = "iat"
+SUBJECT = "sub"
+
+
+class JWTGenerator:
+    """
+    Creates and signs a JWT with the specified private key file, username, and account identifier. The JWTGenerator
+    keeps the generated token and only regenerates the token if a specified period of time has passed.
+    """
+
+    _DEFAULT_LIFETIME = timedelta(minutes=59)  # The tokens will have a 59-minute lifetime
+    _DEFAULT_RENEWAL_DELTA = timedelta(minutes=54)  # Tokens will be renewed after 54 minutes
+    ALGORITHM = "RS256"  # Tokens will be generated using RSA with SHA256
+
+    def __init__(
+        self,
+        account: str,
+        user: str,
+        private_key: types.PRIVATE_KEY_TYPES,
+        lifetime: Optional[timedelta] = None,
+        renewal_delay: Optional[timedelta] = None,
+    ) -> None:
+        """
+        Create a new JWTGenerator object.
+
+        Args:
+            account: The account identifier.
+            user: The username.
+            private_key: The private key used to sign the JWT.
+            lifetime: The lifetime of the token.
+            renewal_delay: The time before the token expires to renew it.
+        """
+
+        # Construct the fully qualified name of the user in uppercase.
+        self.account = JWTGenerator._prepare_account_name_for_jwt(account)
+        self.user = user.upper()
+        self.qualified_username = self.account + "." + self.user
+        self.private_key = private_key
+        self.public_key_fp = JWTGenerator._calculate_public_key_fingerprint(self.private_key)
+
+        self.issuer = self.qualified_username + "." + self.public_key_fp
+        self.lifetime = lifetime or JWTGenerator._DEFAULT_LIFETIME
+        self.renewal_delay = renewal_delay or JWTGenerator._DEFAULT_RENEWAL_DELTA
+        self.renew_time = datetime.now(timezone.utc)
+        self.token: Optional[str] = None
+
+        logger.info(
+            """Creating JWTGenerator with arguments
+            account : %s, user : %s, lifetime : %s, renewal_delay : %s""",
+            self.account,
+            self.user,
+            self.lifetime,
+            self.renewal_delay,
+        )
+
+    @staticmethod
+    def _prepare_account_name_for_jwt(raw_account: str) -> str:
+        account = raw_account
+        if ".global" not in account:
+            # Handle the general case.
+            idx = account.find(".")
+            if idx > 0:
+                account = account[0:idx]
+        else:
+            # Handle the replication case.
+            idx = account.find("-")
+            if idx > 0:
+                account = account[0:idx]
+        # Use uppercase for the account identifier.
+        return account.upper()
+
+    def get_token(self) -> str:
+        now = datetime.now(timezone.utc)  # Fetch the current time
+        if self.token is not None and self.renew_time > now:
+            return self.token
+
+        # If the token has expired or doesn't exist, regenerate the token.
+        logger.info(
+            "Generating a new token because the present time (%s) is later than the renewal time (%s)",
+            now,
+            self.renew_time,
+        )
+        # Calculate the next time we need to renew the token.
+        self.renew_time = now + self.renewal_delay
+
+        # Create our payload
+        payload = {
+            # Set the issuer to the fully qualified username concatenated with the public key fingerprint.
+            ISSUER: self.issuer,
+            # Set the subject to the fully qualified username.
+            SUBJECT: self.qualified_username,
+            # Set the issue time to now.
+            ISSUE_TIME: now,
+            # Set the expiration time, based on the lifetime specified for this object.
+            EXPIRE_TIME: now + self.lifetime,
+        }
+
+        # Regenerate the actual token
+        token = jwt.encode(payload, key=self.private_key, algorithm=JWTGenerator.ALGORITHM)
+        # If you are using a version of PyJWT prior to 2.0, jwt.encode returns a byte string instead of a string.
+        # If the token is a byte string, convert it to a string.
+        if isinstance(token, bytes):
+            token = token.decode("utf-8")
+        self.token = token
+        logger.info(
+            "Generated a JWT with the following payload: %s",
+            jwt.decode(self.token, key=self.private_key.public_key(), algorithms=[JWTGenerator.ALGORITHM]),
+        )
+
+        return token
+
+    @staticmethod
+    def _calculate_public_key_fingerprint(private_key: types.PRIVATE_KEY_TYPES) -> str:
+        # Get the raw bytes of public key.
+        public_key_raw = private_key.public_key().public_bytes(
+            serialization.Encoding.DER, serialization.PublicFormat.SubjectPublicKeyInfo
+        )
+
+        # Get the sha256 hash of the raw bytes.
+        sha256hash = hashlib.sha256()
+        sha256hash.update(public_key_raw)
+
+        # Base64-encode the value and prepend the prefix 'SHA256:'.
+        public_key_fp = "SHA256:" + base64.b64encode(sha256hash.digest()).decode("utf-8")
+        logger.info("Public key fingerprint is %s", public_key_fp)
+
+        return public_key_fp

snowflake/ml/model/_client/model/model_version_impl.py

@@ -14,7 +14,7 @@ from snowflake.ml.model._client.ops import metadata_ops, model_ops, service_ops
 from snowflake.ml.model._model_composer import model_composer
 from snowflake.ml.model._model_composer.model_manifest import model_manifest_schema
 from snowflake.ml.model._packager.model_handlers import snowmlmodel
-from snowflake.snowpark import Session, dataframe
+from snowflake.snowpark import Session, async_job, dataframe
 
 _TELEMETRY_PROJECT = "MLOps"
 _TELEMETRY_SUBPROJECT = "ModelManagement"
@@ -631,7 +631,8 @@ class ModelVersion(lineage_node.LineageNode):
         max_batch_rows: Optional[int] = None,
         force_rebuild: bool = False,
         build_external_access_integration: Optional[str] = None,
-    ) -> str:
+        block: bool = True,
+    ) -> Union[str, async_job.AsyncJob]:
         """Create an inference service with the given spec.
 
         Args:
@@ -659,6 +660,9 @@ class ModelVersion(lineage_node.LineageNode):
             force_rebuild: Whether to force a model inference image rebuild.
             build_external_access_integration: (Deprecated) The external access integration for image build. This is
                 usually permitting access to conda & PyPI repositories.
+            block: A bool value indicating whether this function will wait until the service is available.
+                When it is ``False``, this function executes the underlying service creation asynchronously
+                and returns an :class:`AsyncJob`.
         """
         ...
 
@@ -679,7 +683,8 @@ class ModelVersion(lineage_node.LineageNode):
         max_batch_rows: Optional[int] = None,
         force_rebuild: bool = False,
         build_external_access_integrations: Optional[List[str]] = None,
-    ) -> str:
+        block: bool = True,
+    ) -> Union[str, async_job.AsyncJob]:
         """Create an inference service with the given spec.
 
         Args:
@@ -707,6 +712,9 @@ class ModelVersion(lineage_node.LineageNode):
             force_rebuild: Whether to force a model inference image rebuild.
             build_external_access_integrations: The external access integrations for image build. This is usually
                 permitting access to conda & PyPI repositories.
+            block: A bool value indicating whether this function will wait until the service is available.
+                When it is ``False``, this function executes the underlying service creation asynchronously
+                and returns an :class:`AsyncJob`.
         """
         ...
 
@@ -742,7 +750,8 @@ class ModelVersion(lineage_node.LineageNode):
         force_rebuild: bool = False,
         build_external_access_integration: Optional[str] = None,
         build_external_access_integrations: Optional[List[str]] = None,
-    ) -> str:
+        block: bool = True,
+    ) -> Union[str, async_job.AsyncJob]:
         """Create an inference service with the given spec.
 
         Args:
@@ -772,12 +781,16 @@ class ModelVersion(lineage_node.LineageNode):
                 usually permitting access to conda & PyPI repositories.
             build_external_access_integrations: The external access integrations for image build. This is usually
                 permitting access to conda & PyPI repositories.
+            block: A bool value indicating whether this function will wait until the service is available.
+                When it is False, this function executes the underlying service creation asynchronously
+                and returns an AsyncJob.
 
         Raises:
             ValueError: Illegal external access integration arguments.
 
         Returns:
-            Result information about service creation from server.
+            If `block=True`, return result information about service creation from server.
+            Otherwise, return the service creation AsyncJob.
         """
         statement_params = telemetry.get_statement_params(
             project=_TELEMETRY_PROJECT,
@@ -829,6 +842,7 @@ class ModelVersion(lineage_node.LineageNode):
                 if build_external_access_integrations is None
                 else [sql_identifier.SqlIdentifier(eai) for eai in build_external_access_integrations]
             ),
+            block=block,
             statement_params=statement_params,
         )
 

snowflake/ml/model/_client/ops/service_ops.py

@@ -6,7 +6,7 @@ import re
 import tempfile
 import threading
 import time
-from typing import Any, Dict, List, Optional, Tuple, cast
+from typing import Any, Dict, List, Optional, Tuple, Union, cast
 
 from packaging import version
 
@@ -15,7 +15,7 @@ from snowflake.ml._internal import file_utils
 from snowflake.ml._internal.utils import service_logger, snowflake_env, sql_identifier
 from snowflake.ml.model._client.service import model_deployment_spec
 from snowflake.ml.model._client.sql import service as service_sql, stage as stage_sql
-from snowflake.snowpark import exceptions, row, session
+from snowflake.snowpark import async_job, exceptions, row, session
 from snowflake.snowpark._internal import utils as snowpark_utils
 
 module_logger = service_logger.get_logger(__name__, service_logger.LogColor.GREY)
@@ -107,8 +107,9 @@ class ServiceOperator:
         max_batch_rows: Optional[int],
         force_rebuild: bool,
         build_external_access_integrations: Optional[List[sql_identifier.SqlIdentifier]],
+        block: bool,
         statement_params: Optional[Dict[str, Any]] = None,
-    ) -> str:
+    ) -> Union[str, async_job.AsyncJob]:
 
         # Fall back to the registry's database and schema if not provided
         database_name = database_name or self._database_name
@@ -204,11 +205,15 @@ class ServiceOperator:
         log_thread = self._start_service_log_streaming(
             async_job, services, model_inference_service_exists, force_rebuild, statement_params
         )
-        log_thread.join()
 
-        res = cast(str, cast(List[row.Row], async_job.result())[0][0])
-        module_logger.info(f"Inference service {service_name} deployment complete: {res}")
-        return res
+        if block:
+            log_thread.join()
+
+            res = cast(str, cast(List[row.Row], async_job.result())[0][0])
+            module_logger.info(f"Inference service {service_name} deployment complete: {res}")
+            return res
+        else:
+            return async_job
 
     def _start_service_log_streaming(
         self,

snowflake/ml/model/_client/sql/stage.py

@@ -15,6 +15,6 @@ class StageSQLClient(_base._BaseSQLClient):
     ) -> None:
         query_result_checker.SqlResultValidator(
             self._session,
-            f"CREATE TEMPORARY STAGE {self.fully_qualified_object_name(database_name, schema_name, stage_name)}",
+            f"CREATE SCOPED TEMPORARY STAGE {self.fully_qualified_object_name(database_name, schema_name, stage_name)}",
             statement_params=statement_params,
         ).has_dimensions(expected_rows=1, expected_cols=1).validate()

snowflake/ml/model/_packager/model_handlers/sklearn.py

@@ -164,6 +164,8 @@ class SKLModelHandler(_base.BaseModelHandler[Union["sklearn.base.BaseEstimator",
                         stacklevel=1,
                     )
                     enable_explainability = False
+                elif model_meta.task == model_types.Task.UNKNOWN:
+                    enable_explainability = False
                 else:
                     enable_explainability = True
             if enable_explainability:

snowflake/ml/model/_packager/model_runtime/_snowml_inference_alternative_requirements.py

@@ -1,2 +1,2 @@
-REQUIREMENTS = ['absl-py>=0.15,<2', 'aiohttp!=4.0.0a0, !=4.0.0a1', 'anyio>=3.5.0,<4', 'cachetools>=3.1.1,<6', 'cloudpickle>=2.0.0', 'cryptography', 'fsspec>=2022.11,<2024', 'importlib_resources>=6.1.1, <7', 'numpy>=1.23,<2', 'packaging>=20.9,<25', 'pandas>=1.0.0,<3', 'pyarrow', 'pytimeparse>=1.1.8,<2', 'pyyaml>=6.0,<7', 'requests', 'retrying>=1.3.3,<2', 's3fs>=2022.11,<2024', 'scikit-learn>=1.4,<1.6', 'scipy>=1.9,<2', 'snowflake-connector-python>=3.5.0,<4', 'snowflake-snowpark-python>=1.17.0,<2', 'sqlparse>=0.4,<1', 'typing-extensions>=4.1.0,<5', 'xgboost>=1.7.3,<3']
-ALL_REQUIREMENTS=['absl-py>=0.15,<2', 'aiohttp!=4.0.0a0, !=4.0.0a1', 'anyio>=3.5.0,<4', 'cachetools>=3.1.1,<6', 'catboost>=1.2.0, <2', 'cloudpickle>=2.0.0', 'cryptography', 'fsspec>=2022.11,<2024', 'importlib_resources>=6.1.1, <7', 'lightgbm>=4.1.0, <5', 'mlflow>=2.1.0,<2.4', 'numpy>=1.23,<2', 'packaging>=20.9,<25', 'pandas>=1.0.0,<3', 'pyarrow', 'pytimeparse>=1.1.8,<2', 'pytorch>=2.0.1,<2.3.0', 'pyyaml>=6.0,<7', 'requests', 'retrying>=1.3.3,<2', 's3fs>=2022.11,<2024', 'scikit-learn>=1.4,<1.6', 'scipy>=1.9,<2', 'sentence-transformers>=2.2.2,<3', 'sentencepiece>=0.1.95,<1', 'shap>=0.46.0,<1', 'snowflake-connector-python>=3.5.0,<4', 'snowflake-snowpark-python>=1.17.0,<2', 'sqlparse>=0.4,<1', 'tensorflow>=2.10,<3', 'tokenizers>=0.10,<1', 'torchdata>=0.4,<1', 'transformers>=4.32.1,<5', 'typing-extensions>=4.1.0,<5', 'xgboost>=1.7.3,<3']
+REQUIREMENTS = ['absl-py>=0.15,<2', 'aiohttp!=4.0.0a0, !=4.0.0a1', 'anyio>=3.5.0,<4', 'cachetools>=3.1.1,<6', 'cloudpickle>=2.0.0', 'cryptography', 'fsspec>=2022.11,<2024', 'importlib_resources>=6.1.1, <7', 'numpy>=1.23,<2', 'packaging>=20.9,<25', 'pandas>=1.0.0,<3', 'pyarrow', 'pyjwt>=2.0.0, <3', 'pytimeparse>=1.1.8,<2', 'pyyaml>=6.0,<7', 'requests', 'retrying>=1.3.3,<2', 's3fs>=2022.11,<2024', 'scikit-learn>=1.4,<1.6', 'scipy>=1.9,<2', 'snowflake-connector-python>=3.5.0,<4', 'snowflake-snowpark-python>=1.17.0,<2', 'sqlparse>=0.4,<1', 'typing-extensions>=4.1.0,<5', 'xgboost>=1.7.3,<3']
+ALL_REQUIREMENTS=['absl-py>=0.15,<2', 'aiohttp!=4.0.0a0, !=4.0.0a1', 'anyio>=3.5.0,<4', 'cachetools>=3.1.1,<6', 'catboost>=1.2.0, <2', 'cloudpickle>=2.0.0', 'cryptography', 'fsspec>=2022.11,<2024', 'importlib_resources>=6.1.1, <7', 'lightgbm>=4.1.0, <5', 'mlflow>=2.1.0,<2.4', 'numpy>=1.23,<2', 'packaging>=20.9,<25', 'pandas>=1.0.0,<3', 'pyarrow', 'pyjwt>=2.0.0, <3', 'pytimeparse>=1.1.8,<2', 'pytorch>=2.0.1,<2.3.0', 'pyyaml>=6.0,<7', 'requests', 'retrying>=1.3.3,<2', 's3fs>=2022.11,<2024', 'scikit-learn>=1.4,<1.6', 'scipy>=1.9,<2', 'sentence-transformers>=2.2.2,<3', 'sentencepiece>=0.1.95,<1', 'shap>=0.46.0,<1', 'snowflake-connector-python>=3.5.0,<4', 'snowflake-snowpark-python>=1.17.0,<2', 'sqlparse>=0.4,<1', 'tensorflow>=2.10,<3', 'tokenizers>=0.10,<1', 'torchdata>=0.4,<1', 'transformers>=4.32.1,<5', 'typing-extensions>=4.1.0,<5', 'xgboost>=1.7.3,<3']

snowflake/ml/model/_signatures/utils.py

@@ -118,7 +118,6 @@ def huggingface_pipeline_signature_auto_infer(task: str, params: Dict[str, Any])
             category=DeprecationWarning,
             stacklevel=1,
         )
-
         return core.ModelSignature(
             inputs=[
                 core.FeatureSpec(name="user_inputs", dtype=core.DataType.STRING, shape=(-1,)),

snowflake/ml/monitoring/_client/model_monitor_sql_client.py

@@ -1,6 +1,4 @@
-import typing
-from collections import Counter
-from typing import Any, Dict, List, Mapping, Optional, Set
+from typing import Any, Dict, List, Mapping, Optional
 
 from snowflake import snowpark
 from snowflake.ml._internal.utils import (
@@ -10,27 +8,12 @@ from snowflake.ml._internal.utils import (
     table_manager,
 )
 from snowflake.ml.model._client.sql import _base
-from snowflake.ml.model._model_composer.model_manifest import model_manifest_schema
 from snowflake.snowpark import session, types
 
-SNOWML_MONITORING_METADATA_TABLE_NAME = "_SYSTEM_MONITORING_METADATA"
-
 MODEL_JSON_COL_NAME = "model"
 MODEL_JSON_MODEL_NAME_FIELD = "model_name"
 MODEL_JSON_VERSION_NAME_FIELD = "version_name"
 
-MONITOR_NAME_COL_NAME = "MONITOR_NAME"
-SOURCE_TABLE_NAME_COL_NAME = "SOURCE_TABLE_NAME"
-FQ_MODEL_NAME_COL_NAME = "FULLY_QUALIFIED_MODEL_NAME"
-VERSION_NAME_COL_NAME = "MODEL_VERSION_NAME"
-FUNCTION_NAME_COL_NAME = "FUNCTION_NAME"
-TASK_COL_NAME = "TASK"
-MONITORING_ENABLED_COL_NAME = "IS_ENABLED"
-TIMESTAMP_COL_NAME_COL_NAME = "TIMESTAMP_COLUMN_NAME"
-PREDICTION_COL_NAMES_COL_NAME = "PREDICTION_COLUMN_NAMES"
-LABEL_COL_NAMES_COL_NAME = "LABEL_COLUMN_NAMES"
-ID_COL_NAMES_COL_NAME = "ID_COLUMN_NAMES"
-
 
 def _build_sql_list_from_columns(columns: List[sql_identifier.SqlIdentifier]) -> str:
     sql_list = ", ".join([f"'{column}'" for column in columns])
@@ -146,19 +129,6 @@ class ModelMonitorSQLClient:
             .validate()
         )
 
-    def _validate_unique_columns(
-        self,
-        timestamp_column: sql_identifier.SqlIdentifier,
-        id_columns: List[sql_identifier.SqlIdentifier],
-        prediction_columns: List[sql_identifier.SqlIdentifier],
-        label_columns: List[sql_identifier.SqlIdentifier],
-    ) -> None:
-        all_columns = [*id_columns, *prediction_columns, *label_columns, timestamp_column]
-        num_all_columns = len(all_columns)
-        num_unique_columns = len(set(all_columns))
-        if num_all_columns != num_unique_columns:
-            raise ValueError("Column names must be unique across id, timestamp, prediction, and label columns.")
-
     def validate_existence_by_name(
         self,
         *,
@@ -244,125 +214,6 @@ class ModelMonitorSQLClient:
         if not all([column_name in source_column_schema for column_name in id_columns]):
             raise ValueError(f"ID column(s): {id_columns} do not exist in source.")
 
-    def _validate_timestamp_column_type(
-        self, table_schema: Mapping[str, types.DataType], timestamp_column: sql_identifier.SqlIdentifier
-    ) -> None:
-        """Ensures columns have the same type.
-
-        Args:
-            table_schema: Dictionary of column names and types in the source table.
-            timestamp_column: Name of the timestamp column.
-
-        Raises:
-            ValueError: If the timestamp column is not of type TimestampType.
-        """
-        if not isinstance(table_schema[timestamp_column], types.TimestampType):
-            raise ValueError(
-                f"Timestamp column: {timestamp_column} must be TimestampType. "
-                f"Found: {table_schema[timestamp_column]}"
-            )
-
-    def _validate_id_columns_types(
-        self, table_schema: Mapping[str, types.DataType], id_columns: List[sql_identifier.SqlIdentifier]
-    ) -> None:
-        """Ensures id columns have the correct type.
-
-        Args:
-            table_schema: Dictionary of column names and types in the source table.
-            id_columns: List of id column names.
-
-        Raises:
-            ValueError: If the id column is not of type StringType.
-        """
-        id_column_types = list({table_schema[column_name] for column_name in id_columns})
-        all_id_columns_string = all([isinstance(column_type, types.StringType) for column_type in id_column_types])
-        if not all_id_columns_string:
-            raise ValueError(f"Id columns must all be StringType. Found: {id_column_types}")
-
-    def _validate_prediction_columns_types(
-        self, table_schema: Mapping[str, types.DataType], prediction_columns: List[sql_identifier.SqlIdentifier]
-    ) -> None:
-        """Ensures prediction columns have the same type.
-
-        Args:
-            table_schema: Dictionary of column names and types in the source table.
-            prediction_columns: List of prediction column names.
-
-        Raises:
-            ValueError: If the prediction columns do not share the same type.
-        """
-
-        prediction_column_types = {table_schema[column_name] for column_name in prediction_columns}
-        if len(prediction_column_types) > 1:
-            raise ValueError(f"Prediction column types must be the same. Found: {prediction_column_types}")
-
-    def _validate_label_columns_types(
-        self,
-        table_schema: Mapping[str, types.DataType],
-        label_columns: List[sql_identifier.SqlIdentifier],
-    ) -> None:
-        """Ensures label columns have the same type, and the correct type for the score type.
-
-        Args:
-            table_schema: Dictionary of column names and types in the source table.
-            label_columns: List of label column names.
-
-        Raises:
-            ValueError: If the label columns do not share the same type.
-        """
-        label_column_types = {table_schema[column_name] for column_name in label_columns}
-        if len(label_column_types) > 1:
-            raise ValueError(f"Label column types must be the same. Found: {label_column_types}")
-
-    def _validate_column_types(
-        self,
-        *,
-        table_schema: Mapping[str, types.DataType],
-        timestamp_column: sql_identifier.SqlIdentifier,
-        id_columns: List[sql_identifier.SqlIdentifier],
-        prediction_columns: List[sql_identifier.SqlIdentifier],
-        label_columns: List[sql_identifier.SqlIdentifier],
-    ) -> None:
-        """Ensures columns have the expected type.
-
-        Args:
-            table_schema: Dictionary of column names and types in the source table.
-            timestamp_column: Name of the timestamp column.
-            id_columns: List of id column names.
-            prediction_columns: List of prediction column names.
-            label_columns: List of label column names.
-        """
-        self._validate_timestamp_column_type(table_schema, timestamp_column)
-        self._validate_id_columns_types(table_schema, id_columns)
-        self._validate_prediction_columns_types(table_schema, prediction_columns)
-        self._validate_label_columns_types(table_schema, label_columns)
-        # TODO(SNOW-1646693): Validate label makes sense with model task
-
-    def _validate_source_table_features_shape(
-        self,
-        table_schema: Mapping[str, types.DataType],
-        special_columns: Set[sql_identifier.SqlIdentifier],
-        model_function: model_manifest_schema.ModelFunctionInfo,
-    ) -> None:
-        table_schema_without_special_columns = {
-            k: v for k, v in table_schema.items() if sql_identifier.SqlIdentifier(k) not in special_columns
-        }
-        schema_column_types_to_count: typing.Counter[types.DataType] = Counter()
-        for column_type in table_schema_without_special_columns.values():
-            schema_column_types_to_count[column_type] += 1
-
-        inputs = model_function["signature"].inputs
-        function_input_types = [input.as_snowpark_type() for input in inputs]
-        function_input_types_to_count: typing.Counter[types.DataType] = Counter()
-        for function_input_type in function_input_types:
-            function_input_types_to_count[function_input_type] += 1
-
-        if function_input_types_to_count != schema_column_types_to_count:
-            raise ValueError(
-                "Model function input types do not match the source table input columns types. "
-                f"Model function expected: {inputs} but got {table_schema_without_special_columns}"
-            )
-
     def validate_source(
         self,
         *,
@@ -395,22 +246,6 @@ class ModelMonitorSQLClient:
             id_columns=id_columns,
         )
 
-    def delete_monitor_metadata(
-        self,
-        name: str,
-        statement_params: Optional[Dict[str, Any]] = None,
-    ) -> None:
-        """Delete the row in the metadata table corresponding to the given monitor name.
-
-        Args:
-            name: Name of the model monitor whose metadata should be deleted.
-            statement_params: Optional set of statement_params to include with query.
-        """
-        self._sql_client._session.sql(
-            f"""DELETE FROM {self._database_name}.{self._schema_name}.{SNOWML_MONITORING_METADATA_TABLE_NAME}
-            WHERE {MONITOR_NAME_COL_NAME} = '{name}'""",
-        ).collect(statement_params=statement_params)
-
     def _alter_monitor(
         self,
         operation: str,

snowflake/ml/monitoring/_manager/model_monitor_manager.py

@@ -14,15 +14,6 @@ from snowflake.snowpark import session
 class ModelMonitorManager:
     """Class to manage internal operations for Model Monitor workflows."""
 
-    def _validate_task_from_model_version(
-        self,
-        model_version: model_version_impl.ModelVersion,
-    ) -> type_hints.Task:
-        task = model_version.get_model_task()
-        if task == type_hints.Task.UNKNOWN:
-            raise ValueError("Registry model must be logged with task in order to be monitored.")
-        return task
-
     def __init__(
         self,
         session: session.Session,
@@ -51,6 +42,15 @@ class ModelMonitorManager:
             schema_name=self._schema_name,
         )
 
+    def _validate_task_from_model_version(
+        self,
+        model_version: model_version_impl.ModelVersion,
+    ) -> type_hints.Task:
+        task = model_version.get_model_task()
+        if task == type_hints.Task.UNKNOWN:
+            raise ValueError("Registry model must be logged with task in order to be monitored.")
+        return task
+
     def _validate_model_function_from_model_version(
         self, function: str, model_version: model_version_impl.ModelVersion
     ) -> None:

snowflake/ml/monitoring/entities/model_monitor_config.py

@@ -6,23 +6,49 @@ from snowflake.ml.model._client.model import model_version_impl
 
 @dataclass
 class ModelMonitorSourceConfig:
+    """Configuration for the source of data to be monitored."""
+
     source: str
+    """Name of table or view containing monitoring data."""
+
     timestamp_column: str
+    """Name of column in the source containing timestamp."""
+
     id_columns: List[str]
+    """List of columns in the source containing unique identifiers."""
+
     prediction_score_columns: Optional[List[str]] = None
+    """List of columns in the source containing prediction scores.
+    Can be regression scores for regression models and probability scores for classification models."""
+
     prediction_class_columns: Optional[List[str]] = None
+    """List of columns in the source containing prediction classes for classification models."""
+
     actual_score_columns: Optional[List[str]] = None
+    """List of columns in the source containing actual scores."""
+
     actual_class_columns: Optional[List[str]] = None
+    """List of columns in the source containing actual classes for classification models."""
+
     baseline: Optional[str] = None
+    """Name of table containing the baseline data."""
 
 
 @dataclass
 class ModelMonitorConfig:
+    """Configuration for the Model Monitor."""
+
     model_version: model_version_impl.ModelVersion
+    """Model version to monitor."""
 
-    # Python model function name
     model_function_name: str
+    """Function name in the model to monitor."""
+
     background_compute_warehouse_name: str
-    # TODO: Add support for pythonic notion of time.
+    """Name of the warehouse to use for background compute."""
+
     refresh_interval: str = "1 hour"
+    """Interval at which to refresh the monitoring data."""
+
     aggregation_window: str = "1 day"
+    """Window for aggregating monitoring data."""

snowflake/ml/monitoring/model_monitor.py

@@ -1,5 +1,7 @@
+from snowflake import snowpark
 from snowflake.ml._internal import telemetry
 from snowflake.ml._internal.utils import sql_identifier
+from snowflake.ml.monitoring import model_monitor_version
 from snowflake.ml.monitoring._client import model_monitor_sql_client
 
 
@@ -9,13 +11,8 @@ class ModelMonitor:
     name: sql_identifier.SqlIdentifier
     _model_monitor_client: model_monitor_sql_client.ModelMonitorSQLClient
 
-    statement_params = telemetry.get_statement_params(
-        telemetry.TelemetryProject.MLOPS.value,
-        telemetry.TelemetrySubProject.MONITORING.value,
-    )
-
     def __init__(self) -> None:
-        raise RuntimeError("ModelMonitor's initializer is not meant to be used.")
+        raise RuntimeError("Model Monitor's initializer is not meant to be used.")
 
     @classmethod
     def _ref(
@@ -28,10 +25,28 @@ class ModelMonitor:
         self._model_monitor_client = model_monitor_client
         return self
 
+    @telemetry.send_api_usage_telemetry(
+        project=telemetry.TelemetryProject.MLOPS.value,
+        subproject=telemetry.TelemetrySubProject.MONITORING.value,
+    )
+    @snowpark._internal.utils.private_preview(version=model_monitor_version.SNOWFLAKE_ML_MONITORING_MIN_VERSION)
     def suspend(self) -> None:
-        """Suspend pipeline for ModelMonitor"""
-        self._model_monitor_client.suspend_monitor(self.name, statement_params=self.statement_params)
-
+        """Suspend the Model Monitor"""
+        statement_params = telemetry.get_statement_params(
+            telemetry.TelemetryProject.MLOPS.value,
+            telemetry.TelemetrySubProject.MONITORING.value,
+        )
+        self._model_monitor_client.suspend_monitor(self.name, statement_params=statement_params)
+
+    @telemetry.send_api_usage_telemetry(
+        project=telemetry.TelemetryProject.MLOPS.value,
+        subproject=telemetry.TelemetrySubProject.MONITORING.value,
+    )
+    @snowpark._internal.utils.private_preview(version=model_monitor_version.SNOWFLAKE_ML_MONITORING_MIN_VERSION)
     def resume(self) -> None:
-        """Resume pipeline for ModelMonitor"""
-        self._model_monitor_client.resume_monitor(self.name, statement_params=self.statement_params)
+        """Resume the Model Monitor"""
+        statement_params = telemetry.get_statement_params(
+            telemetry.TelemetryProject.MLOPS.value,
+            telemetry.TelemetrySubProject.MONITORING.value,
+        )
+        self._model_monitor_client.resume_monitor(self.name, statement_params=statement_params)

snowflake/ml/registry/registry.py

@@ -388,15 +388,15 @@ class Registry:
         source_config: model_monitor_config.ModelMonitorSourceConfig,
         model_monitor_config: model_monitor_config.ModelMonitorConfig,
     ) -> model_monitor.ModelMonitor:
-        """Add a Model Monitor to the Registry
+        """Add a Model Monitor to the Registry.
 
         Args:
-            name: Name of Model Monitor to create
-            source_config: Configuration options of table for ModelMonitor.
-            model_monitor_config: Configuration options of ModelMonitor.
+            name: Name of Model Monitor to create.
+            source_config: Configuration options of table for Model Monitor.
+            model_monitor_config: Configuration options of Model Monitor.
 
         Returns:
-            The newly added ModelMonitor object.
+            The newly added Model Monitor object.
 
         Raises:
             ValueError: If monitoring is not enabled in the Registry.
@@ -407,16 +407,16 @@ class Registry:
 
     @overload
     def get_monitor(self, model_version: model_version_impl.ModelVersion) -> model_monitor.ModelMonitor:
-        """Get a Model Monitor on a ModelVersion from the Registry
+        """Get a Model Monitor on a Model Version from the Registry.
 
         Args:
-            model_version: ModelVersion for which to retrieve the ModelMonitor.
+            model_version: Model Version for which to retrieve the Model Monitor.
         """
         ...
 
     @overload
     def get_monitor(self, name: str) -> model_monitor.ModelMonitor:
-        """Get a Model Monitor from the Registry
+        """Get a Model Monitor by name from the Registry.
 
         Args:
             name: Name of Model Monitor to retrieve.
@@ -431,14 +431,14 @@ class Registry:
     def get_monitor(
         self, *, name: Optional[str] = None, model_version: Optional[model_version_impl.ModelVersion] = None
     ) -> model_monitor.ModelMonitor:
-        """Get a Model Monitor from the Registry
+        """Get a Model Monitor from the Registry.
 
         Args:
             name: Name of Model Monitor to retrieve.
-            model_version: ModelVersion for which to retrieve the ModelMonitor.
+            model_version: Model Version for which to retrieve the Model Monitor.
 
         Returns:
-            The fetched ModelMonitor.
+            The fetched Model Monitor.
 
         Raises:
             ValueError: If monitoring is not enabled in the Registry.
@@ -476,7 +476,7 @@ class Registry:
     )
     @snowpark._internal.utils.private_preview(version=model_monitor_version.SNOWFLAKE_ML_MONITORING_MIN_VERSION)
     def delete_monitor(self, name: str) -> None:
-        """Delete a Model Monitor from the Registry
+        """Delete a Model Monitor by name from the Registry.
 
         Args:
             name: Name of the Model Monitor to delete.

snowflake/ml/utils/authentication.py

@@ -0,0 +1,75 @@
+import http
+import logging
+from datetime import timedelta
+from typing import Dict, Optional
+
+import requests
+from cryptography.hazmat.primitives.asymmetric import types
+from requests import auth
+
+from snowflake.ml._internal.utils import jwt_generator
+
+logger = logging.getLogger(__name__)
+_JWT_TOKEN_CACHE: Dict[str, Dict[int, str]] = {}
+
+
+def get_jwt_token_generator(
+    account: str,
+    user: str,
+    private_key: types.PRIVATE_KEY_TYPES,
+    lifetime: Optional[timedelta] = None,
+    renewal_delay: Optional[timedelta] = None,
+) -> jwt_generator.JWTGenerator:
+    return jwt_generator.JWTGenerator(account, user, private_key, lifetime=lifetime, renewal_delay=renewal_delay)
+
+
+def _get_snowflake_token_by_jwt(
+    jwt_token_generator: jwt_generator.JWTGenerator,
+    account: Optional[str] = None,
+    role: Optional[str] = None,
+    endpoint: Optional[str] = None,
+    snowflake_account_url: Optional[str] = None,
+) -> str:
+    scope_role = f"session:role:{role}" if role is not None else None
+    scope = " ".join(filter(None, [scope_role, endpoint]))
+    data = {
+        "grant_type": "urn:ietf:params:oauth:grant-type:jwt-bearer",
+        "scope": scope or None,
+        "assertion": jwt_token_generator.get_token(),
+    }
+    account = account or jwt_token_generator.account
+    url = f"https://{account}.snowflakecomputing.com/oauth/token"
+    if snowflake_account_url:
+        url = f"{snowflake_account_url}/oauth/token"
+
+    cache_key = hash(frozenset(data.items()))
+    if url in _JWT_TOKEN_CACHE:
+        if cache_key in _JWT_TOKEN_CACHE[url]:
+            return _JWT_TOKEN_CACHE[url][cache_key]
+    else:
+        _JWT_TOKEN_CACHE[url] = {}
+
+    response = requests.post(url, data=data)
+    if response.status_code != http.HTTPStatus.OK:
+        raise RuntimeError(f"Failed to get snowflake token: {response.status_code} {response.content!r}")
+    auth_token = response.text
+    _JWT_TOKEN_CACHE[url][cache_key] = auth_token
+    return auth_token
+
+
+class SnowflakeJWTTokenAuth(auth.AuthBase):
+    def __init__(
+        self,
+        jwt_token_generator: jwt_generator.JWTGenerator,
+        account: Optional[str] = None,
+        role: Optional[str] = None,
+        endpoint: Optional[str] = None,
+        snowflake_account_url: Optional[str] = None,
+    ) -> None:
+        self.snowflake_token = _get_snowflake_token_by_jwt(
+            jwt_token_generator, account, role, endpoint, snowflake_account_url
+        )
+
+    def __call__(self, r: requests.PreparedRequest) -> requests.PreparedRequest:
+        r.headers["Authorization"] = f'Snowflake Token="{self.snowflake_token}"'
+        return r

snowflake/ml/version.py

@@ -1 +1 @@
-VERSION="1.7.1"
+VERSION="1.7.2"

{snowflake_ml_python-1.7.1.dist-info → snowflake_ml_python-1.7.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: snowflake-ml-python
-Version: 1.7.1
+Version: 1.7.2
 Summary: The machine learning client library that is used for interacting with Snowflake to build machine learning solutions.
 Author-email: "Snowflake, Inc" <support@snowflake.com>
 License: 
@@ -232,61 +232,62 @@ Classifier: Topic :: Scientific/Engineering :: Information Analysis
 Requires-Python: <3.12,>=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
-Requires-Dist: absl-py <2,>=0.15
-Requires-Dist: anyio <4,>=3.5.0
-Requires-Dist: cachetools <6,>=3.1.1
-Requires-Dist: cloudpickle >=2.0.0
+Requires-Dist: absl-py<2,>=0.15
+Requires-Dist: anyio<4,>=3.5.0
+Requires-Dist: cachetools<6,>=3.1.1
+Requires-Dist: cloudpickle>=2.0.0
 Requires-Dist: cryptography
-Requires-Dist: fsspec[http] <2024,>=2022.11
-Requires-Dist: importlib-resources <7,>=6.1.1
-Requires-Dist: numpy <2,>=1.23
-Requires-Dist: packaging <25,>=20.9
-Requires-Dist: pandas <3,>=1.0.0
+Requires-Dist: fsspec[http]<2024,>=2022.11
+Requires-Dist: importlib_resources<7,>=6.1.1
+Requires-Dist: numpy<2,>=1.23
+Requires-Dist: packaging<25,>=20.9
+Requires-Dist: pandas<3,>=1.0.0
 Requires-Dist: pyarrow
-Requires-Dist: pytimeparse <2,>=1.1.8
-Requires-Dist: pyyaml <7,>=6.0
-Requires-Dist: retrying <2,>=1.3.3
-Requires-Dist: s3fs <2024,>=2022.11
-Requires-Dist: scikit-learn <1.6,>=1.4
-Requires-Dist: scipy <2,>=1.9
-Requires-Dist: snowflake-connector-python[pandas] <4,>=3.5.0
-Requires-Dist: snowflake-snowpark-python <2,>=1.17.0
-Requires-Dist: sqlparse <1,>=0.4
-Requires-Dist: typing-extensions <5,>=4.1.0
-Requires-Dist: xgboost <3,>=1.7.3
+Requires-Dist: pyjwt<3,>=2.0.0
+Requires-Dist: pytimeparse<2,>=1.1.8
+Requires-Dist: pyyaml<7,>=6.0
+Requires-Dist: retrying<2,>=1.3.3
+Requires-Dist: s3fs<2024,>=2022.11
+Requires-Dist: scikit-learn<1.6,>=1.4
+Requires-Dist: scipy<2,>=1.9
+Requires-Dist: snowflake-connector-python[pandas]<4,>=3.5.0
+Requires-Dist: snowflake-snowpark-python<2,>=1.17.0
+Requires-Dist: sqlparse<1,>=0.4
+Requires-Dist: typing-extensions<5,>=4.1.0
+Requires-Dist: xgboost<3,>=1.7.3
 Provides-Extra: all
-Requires-Dist: catboost <2,>=1.2.0 ; extra == 'all'
-Requires-Dist: lightgbm <5,>=4.1.0 ; extra == 'all'
-Requires-Dist: mlflow <2.4,>=2.1.0 ; extra == 'all'
-Requires-Dist: peft <1,>=0.5.0 ; extra == 'all'
-Requires-Dist: sentence-transformers <3,>=2.2.2 ; extra == 'all'
-Requires-Dist: sentencepiece <1,>=0.1.95 ; extra == 'all'
-Requires-Dist: shap <1,>=0.46.0 ; extra == 'all'
-Requires-Dist: tensorflow <3,>=2.10 ; extra == 'all'
-Requires-Dist: tokenizers <1,>=0.10 ; extra == 'all'
-Requires-Dist: torch <2.3.0,>=2.0.1 ; extra == 'all'
-Requires-Dist: torchdata <1,>=0.4 ; extra == 'all'
-Requires-Dist: transformers <5,>=4.32.1 ; extra == 'all'
+Requires-Dist: catboost<2,>=1.2.0; extra == "all"
+Requires-Dist: lightgbm<5,>=4.1.0; extra == "all"
+Requires-Dist: mlflow<2.4,>=2.1.0; extra == "all"
+Requires-Dist: peft<1,>=0.5.0; extra == "all"
+Requires-Dist: sentence-transformers<3,>=2.2.2; extra == "all"
+Requires-Dist: sentencepiece<1,>=0.1.95; extra == "all"
+Requires-Dist: shap<1,>=0.46.0; extra == "all"
+Requires-Dist: tensorflow<3,>=2.10; extra == "all"
+Requires-Dist: tokenizers<1,>=0.10; extra == "all"
+Requires-Dist: torch<2.3.0,>=2.0.1; extra == "all"
+Requires-Dist: torchdata<1,>=0.4; extra == "all"
+Requires-Dist: transformers<5,>=4.32.1; extra == "all"
 Provides-Extra: catboost
-Requires-Dist: catboost <2,>=1.2.0 ; extra == 'catboost'
+Requires-Dist: catboost<2,>=1.2.0; extra == "catboost"
 Provides-Extra: lightgbm
-Requires-Dist: lightgbm <5,>=4.1.0 ; extra == 'lightgbm'
+Requires-Dist: lightgbm<5,>=4.1.0; extra == "lightgbm"
 Provides-Extra: llm
-Requires-Dist: peft <1,>=0.5.0 ; extra == 'llm'
+Requires-Dist: peft<1,>=0.5.0; extra == "llm"
 Provides-Extra: mlflow
-Requires-Dist: mlflow <2.4,>=2.1.0 ; extra == 'mlflow'
+Requires-Dist: mlflow<2.4,>=2.1.0; extra == "mlflow"
 Provides-Extra: shap
-Requires-Dist: shap <1,>=0.46.0 ; extra == 'shap'
+Requires-Dist: shap<1,>=0.46.0; extra == "shap"
 Provides-Extra: tensorflow
-Requires-Dist: tensorflow <3,>=2.10 ; extra == 'tensorflow'
+Requires-Dist: tensorflow<3,>=2.10; extra == "tensorflow"
 Provides-Extra: torch
-Requires-Dist: torch <2.3.0,>=2.0.1 ; extra == 'torch'
-Requires-Dist: torchdata <1,>=0.4 ; extra == 'torch'
+Requires-Dist: torch<2.3.0,>=2.0.1; extra == "torch"
+Requires-Dist: torchdata<1,>=0.4; extra == "torch"
 Provides-Extra: transformers
-Requires-Dist: sentence-transformers <3,>=2.2.2 ; extra == 'transformers'
-Requires-Dist: sentencepiece <1,>=0.1.95 ; extra == 'transformers'
-Requires-Dist: tokenizers <1,>=0.10 ; extra == 'transformers'
-Requires-Dist: transformers <5,>=4.32.1 ; extra == 'transformers'
+Requires-Dist: sentence-transformers<3,>=2.2.2; extra == "transformers"
+Requires-Dist: sentencepiece<1,>=0.1.95; extra == "transformers"
+Requires-Dist: tokenizers<1,>=0.10; extra == "transformers"
+Requires-Dist: transformers<5,>=4.32.1; extra == "transformers"
 
 # Snowpark ML
 
@@ -302,7 +303,7 @@ and deployment process, and includes two key components.
 
 ### Snowpark ML Development
 
-[Snowpark ML Development](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowpark-ml-development)
+[Snowpark ML Development](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#ml-modeling)
 provides a collection of python APIs enabling efficient ML model development directly in Snowflake:
 
 1. Modeling API (`snowflake.ml.modeling`) for data preprocessing, feature engineering and model training in Snowflake.
@@ -316,14 +317,21 @@ their native data loader formats.
 1. FileSet API: FileSet provides a Python fsspec-compliant API for materializing data into a Snowflake internal stage
 from a query or Snowpark Dataframe along with a number of convenience APIs.
 
-### Snowpark Model Management [Public Preview]
+### Snowflake MLOps
 
-[Snowpark Model Management](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowpark-ml-ops) complements
-the Snowpark ML Development API, and provides model management capabilities along with integrated deployment into Snowflake.
+Snowflake MLOps contains suit of tools and objects to make ML development cycle. It complements
+the Snowpark ML Development API, and provides end to end development to deployment within Snowflake.
 Currently, the API consists of:
 
-1. Registry: A python API for managing models within Snowflake which also supports deployment of ML models into Snowflake
-as native MODEL object running with Snowflake Warehouse.
+1. [Registry](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowflake-model-registry): A python API
+  allows secure deployment and management of models in Snowflake, supporting models trained both inside and outside of
+  Snowflake.
+2. [Feature Store](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#snowflake-feature-store): A fully
+  integrated solution for defining, managing, storing and discovering ML features derived from your data. The
+  Snowflake Feature Store supports automated, incremental refresh from batch and streaming data sources, so that
+  feature pipelines need be defined only once to be continuously updated with new data.
+3. [Datasets](https://docs.snowflake.com/developer-guide/snowflake-ml/overview#snowflake-datasets): Dataset provide an
+  immutable, versioned snapshot of your data suitable for ingestion by your machine learning models.
 
 ## Getting started
 
@@ -371,9 +379,39 @@ conda install \
 Note that until a `snowflake-ml-python` package version is available in the official Snowflake conda channel, there may
 be compatibility issues. Server-side functionality that `snowflake-ml-python` depends on may not yet be released.
 
+### Verifying the package
+
+1. Install cosign.
+  This example is using golang installation: [installing-cosign-with-go](https://edu.chainguard.dev/open-source/sigstore/cosign/how-to-install-cosign/#installing-cosign-with-go).
+1. Download the file from the repository like [pypi](https://pypi.org/project/snowflake-ml-python/#files).
+1. Download the signature files from the [release tag](https://github.com/snowflakedb/snowflake-ml-python/releases/tag/1.7.0).
+1. Verify signature on projects signed using Jenkins job:
+
+   ```sh
+   cosign verify-blob snowflake_ml_python-1.7.0.tar.gz --key snowflake-ml-python-1.7.0.pub --signature resources.linux.snowflake_ml_python-1.7.0.tar.gz.sig
+
+   cosign verify-blob snowflake_ml_python-1.7.0.tar.gz --key snowflake-ml-python-1.7.0.pub --signature resources.linux.snowflake_ml_python-1.7.0
+   ```
+
+NOTE: Version 1.7.0 is used as example here. Please choose the the latest version.
+
 # Release History
 
-## 1.7.1
+## 1.7.2
+
+### Bug Fixes
+
+- Model Explainability: Fix issue that explain is enabled for scikit-learn pipeline
+whose task is UNKNOWN and fails later when invoked.
+
+### Behavior Changes
+
+### New Features
+
+- Registry: Support asynchronous model inference service creation with the `block` option
+  in `ModelVersion.create_service()` set to True by default.
+
+## 1.7.1 (2024-11-05)
 
 ### Bug Fixes
 

{snowflake_ml_python-1.7.1.dist-info → snowflake_ml_python-1.7.2.dist-info}/RECORD

@@ -10,7 +10,7 @@ snowflake/cortex/_sse_client.py,sha256=sLYgqAfTOPADCnaWH2RWAJi8KbU_7gSRsTUDcDD5T
 snowflake/cortex/_summarize.py,sha256=bwpFBzBGmNQSoJqKs3IB5wASjAREnC5ZnViSuZK5IrU,1059
 snowflake/cortex/_translate.py,sha256=69YUps6mnhzVdubdU_H0IfUAlbBwF9OPemFEQ34P-ts,1404
 snowflake/cortex/_util.py,sha256=cwRGgrcUo3E05ZaIDT9436vXLQ7GfuBVAjR0QeQ2bDE,3320
-snowflake/ml/version.py,sha256=QyWKL6Zvq-VDoZgBZ32iGHIzxeVh0z4fIKkiZHSX7t4,16
+snowflake/ml/version.py,sha256=wJaJaqPpO6Ic3Pl_5e81zlGKYqi1rf5q8V10jTUEDjA,16
 snowflake/ml/_internal/env.py,sha256=kCrJTRnqQ97VGUVI1cWUPD8HuBWeL5vOOtwUR0NB9Mg,161
 snowflake/ml/_internal/env_utils.py,sha256=J_jitp8jvDoC3a79EbMSDatFRYw-HiXaI9vR81bhtU8,28075
 snowflake/ml/_internal/file_utils.py,sha256=OyXHv-UcItiip1YgLnab6etonUQkYuyDtmplZA0CaoU,13622
@@ -36,6 +36,7 @@ snowflake/ml/_internal/utils/db_utils.py,sha256=HBAY0-XHzCP4ai5q3Yqd8O19Ar_Q9J3x
 snowflake/ml/_internal/utils/formatting.py,sha256=PswZ6Xas7sx3Ok1MBLoH2o7nfXOxaJqpUPg_UqXrQb8,3676
 snowflake/ml/_internal/utils/identifier.py,sha256=fUYXjXKXAkjLUZpomneMHo2wR4_ZNP4ak-5OJxeUS-g,12467
 snowflake/ml/_internal/utils/import_utils.py,sha256=iUIROZdiTGy73UCGpG0N-dKtK54H0ymNVge_QNQYY3A,3220
+snowflake/ml/_internal/utils/jwt_generator.py,sha256=bj7Ltnw68WjRcxtV9t5xrTRvV5ETnvovB-o3Y8QWNBg,5357
 snowflake/ml/_internal/utils/parallelize.py,sha256=Q6_-P2t4DoYNO8DyC1kOl7H3qNL-bUK6EgtlQ_b5ThY,4534
 snowflake/ml/_internal/utils/pkg_version_utils.py,sha256=FwdLHFhxi3CAQQduGjFavEBmkD9Ra6ZTkt6Eub-WoSA,5168
 snowflake/ml/_internal/utils/query_result_checker.py,sha256=h1nbUImdB9lSNCON3uIA0xCm8_JrS-TE-jQXJJs9WfU,10668
@@ -101,17 +102,17 @@ snowflake/ml/model/custom_model.py,sha256=O60mjz2Vy8A0Rt3obq43zBT3BxkU7CIcN0AkHs
 snowflake/ml/model/model_signature.py,sha256=gZnZPs9zTCYkeFoiQzoGUQYZMydYjzH-4xPTzfqt4hU,30496
 snowflake/ml/model/type_hints.py,sha256=9GPwEuG6B6GSWOXdOy8B1Swz6yDngL865yEtJMd0v1U,8883
 snowflake/ml/model/_client/model/model_impl.py,sha256=pqjK8mSZIQJ_30tRWWFPIo8X35InSVoAunXlQNtSJEM,15369
-snowflake/ml/model/_client/model/model_version_impl.py,sha256=PTVqTkNm1adHUjTTWsUlnTSPiMQV-PZLEaj9UstICqk,39076
+snowflake/ml/model/_client/model/model_version_impl.py,sha256=tGfSR4dF8okdBPeAu7yWVSLtwvnvhnJr9xalKbQZw5M,40144
 snowflake/ml/model/_client/ops/metadata_ops.py,sha256=7cGx8zYzye2_cvZnyGxoukPtT6Q-Kexd-s4yeZmpmj8,4890
 snowflake/ml/model/_client/ops/model_ops.py,sha256=didFBsjb7KJYV_586TUK4c9DudVQvjzlphEXJW0AnmY,43935
-snowflake/ml/model/_client/ops/service_ops.py,sha256=LLvRqBBwyJsjNfphN_VdH8O1aQEPNf97Wmco5dfLUN0,19093
+snowflake/ml/model/_client/ops/service_ops.py,sha256=t_yLtHlAzHc28XDZ543yAALY5iVsRwVw4i9mtiPaXpQ,19237
 snowflake/ml/model/_client/service/model_deployment_spec.py,sha256=uyh5k_u8mVP5T4lf0jq8s2cFuiTsbV_nJL6z1Zum2rM,4456
 snowflake/ml/model/_client/service/model_deployment_spec_schema.py,sha256=eaulF6OFNuDfQz3oPYlDjP26Ww2jWWatm81dCbg602E,825
 snowflake/ml/model/_client/sql/_base.py,sha256=Qrm8M92g3MHb-QnSLUlbd8iVKCRxLhG_zr5M2qmXwJ8,1473
 snowflake/ml/model/_client/sql/model.py,sha256=o36oPq4aU9TwahqY2uODYvICxmj1orLztijJ0yMbWnM,5852
 snowflake/ml/model/_client/sql/model_version.py,sha256=hNMlmwN5JQngKuaeUYV2Bli73RMnHmVH01ABX9NBHFk,20686
 snowflake/ml/model/_client/sql/service.py,sha256=fvQRhRGU4FBeOBouIoQByTvfQg-qbEQKplCG99BPmL0,10408
-snowflake/ml/model/_client/sql/stage.py,sha256=hrCh9P9F4l5R0hLr2r-wLDIEc4XYHMFdX1wNRveMVt0,819
+snowflake/ml/model/_client/sql/stage.py,sha256=165vyAtrScSQWJB8wLXKRUO1QvHTWDmPykeWOyxrDRg,826
 snowflake/ml/model/_client/sql/tag.py,sha256=pwwrcyPtSnkUfDzL3M8kqM0KSx7CaTtgty3HDhVC9vg,4345
 snowflake/ml/model/_model_composer/model_composer.py,sha256=535ElL3Kw8eoUjL7fHd-K20eDCBqvJFwowUx2_UOCl8,6712
 snowflake/ml/model/_model_composer/model_manifest/model_manifest.py,sha256=X6-cKLBZ1X2liIjWnyrd9efQaQhwIoxRSE90Zs0kAZo,7822
@@ -133,7 +134,7 @@ snowflake/ml/model/_packager/model_handlers/lightgbm.py,sha256=E0667G5FFfMssaXjk
 snowflake/ml/model/_packager/model_handlers/mlflow.py,sha256=A3HnCa065jtHsRM40ZxfLv5alk0RYhVmsU4Jt2klRwQ,9189
 snowflake/ml/model/_packager/model_handlers/pytorch.py,sha256=DDcf85xisPLT1PyXdmPrjJpIIepkdmWNXCOpT_dCncw,8294
 snowflake/ml/model/_packager/model_handlers/sentence_transformers.py,sha256=f21fJw2wPsXzzhv71Gi1eHctSlyJ6NAR1EQX5iUL5M8,9842
-snowflake/ml/model/_packager/model_handlers/sklearn.py,sha256=UbtqgOztM9racr_N-SPRymEpUwhZGKov5iv6dcbINy8,13995
+snowflake/ml/model/_packager/model_handlers/sklearn.py,sha256=dwwETBdJJM3AVfl3R6VvvVOZQHgnwIuk9dUUCDOs-w0,14111
 snowflake/ml/model/_packager/model_handlers/snowmlmodel.py,sha256=uhsJ3zK24aavBRO5gNyxv8BHqU9n1TPUBYm1qHTuaxE,12176
 snowflake/ml/model/_packager/model_handlers/tensorflow.py,sha256=SkbnvkElK4UIMgygv9EK9f5hBxWZ2YDroymUC9uBsBk,9169
 snowflake/ml/model/_packager/model_handlers/torchscript.py,sha256=BIdRINO1xZ5uHrR9uA0vExWQymOryTaSpyAMpCCtz8U,8036
@@ -146,7 +147,7 @@ snowflake/ml/model/_packager/model_meta/model_meta_schema.py,sha256=5Sdh1_NCKycL
 snowflake/ml/model/_packager/model_meta_migrator/base_migrator.py,sha256=SORlqpPbOeBg6dvJ3DidHeLVi0w9YF0Zv4tC0Kbc20g,1311
 snowflake/ml/model/_packager/model_meta_migrator/migrator_plans.py,sha256=nf6PWDH_gvX_OiS4A-G6BzyCLFEG4dASU0t5JTsijM4,1041
 snowflake/ml/model/_packager/model_meta_migrator/migrator_v1.py,sha256=qEPzdCw_FzExMbPuyFHupeWlYD88yejLdcmkPwjJzDk,2070
-snowflake/ml/model/_packager/model_runtime/_snowml_inference_alternative_requirements.py,sha256=5YHbTmgPdURGQDZwzmC7mlYSl8q_e7hzHJ-JyMXgDFY,1419
+snowflake/ml/model/_packager/model_runtime/_snowml_inference_alternative_requirements.py,sha256=eR_qxEwsmzaeaRYH9K4wUAG7bhpqZvn07en2vfRV4c4,1459
 snowflake/ml/model/_packager/model_runtime/model_runtime.py,sha256=G52nrjzcZiWBJaed6Z1qKq-HjqtnG2MnywDdU9lPusg,5051
 snowflake/ml/model/_packager/model_task/model_task_utils.py,sha256=0aEUfg71bP5-RkwmzOJBe51yHxLRrtM17tUBoCiuMMk,6310
 snowflake/ml/model/_signatures/base_handler.py,sha256=WwBfe-83Y0m-HcDx1YSYCGwanIe0fb2MWhTeXc1IeJI,1304
@@ -157,7 +158,7 @@ snowflake/ml/model/_signatures/pandas_handler.py,sha256=ACv8egyiK2Sug8uhkQqMDGTT
 snowflake/ml/model/_signatures/pytorch_handler.py,sha256=yEU-V_WRjE8Q7NdHyghl0iYpMiIDzGaIR5Pd_ixB1Hk,4631
 snowflake/ml/model/_signatures/snowpark_handler.py,sha256=2_AY1ssucMICKSPeDjf3mV4WT5farKYdnYkHsvhHZ20,6066
 snowflake/ml/model/_signatures/tensorflow_handler.py,sha256=9bUbxtHpl4kEoFzeDJF87bQPb8RdLLm9OV23-aUyW3s,6114
-snowflake/ml/model/_signatures/utils.py,sha256=-RuAFPJn8JHh8QUMLAgMbgpuDvNLI6gVDeLf-lvUBxQ,13109
+snowflake/ml/model/_signatures/utils.py,sha256=1E_mV1qdUuob8tjB8WaOEfuo2rmQ2FtOgTNyXZGzoJg,13108
 snowflake/ml/model/models/huggingface_pipeline.py,sha256=62GpPZxBheqCnFNxNOggiDE1y9Dhst-v6D4IkGLuDeQ,10221
 snowflake/ml/modeling/_internal/constants.py,sha256=aJGngY599w3KqN8cDZCYrjbWe6UwYIbgv0gx0Ukdtc0,105
 snowflake/ml/modeling/_internal/estimator_utils.py,sha256=mbMm8_5tQde_sQDwI8pS3ljHZ8maCHl2Shb5nQwLYac,11872
@@ -379,23 +380,23 @@ snowflake/ml/modeling/xgboost/xgb_classifier.py,sha256=2QEK6-NihXjKXO8Ue-fOZDyuc
 snowflake/ml/modeling/xgboost/xgb_regressor.py,sha256=ZorEmRohT2-AUdS8fK0xH8BdB8ENxvVMMDYy34Jzm1o,61703
 snowflake/ml/modeling/xgboost/xgbrf_classifier.py,sha256=67jh9RosrTeYCWsJbnJ6_MQICHeG22z-DMy8CegP8Vg,62383
 snowflake/ml/modeling/xgboost/xgbrf_regressor.py,sha256=7_ZwF_QvVqBrkFx_zgGgLXyxtbX26XrWWLozAF-EBB0,61908
-snowflake/ml/monitoring/model_monitor.py,sha256=p8FWMpr9O8ScL_y6wdrMUstlpA43gJ0Qiv2e8w-ADts,1374
+snowflake/ml/monitoring/model_monitor.py,sha256=8vJf1YROmJgBLUtpaH-lGKSSJv9R7PxPaQnOdr_j5YE,2200
 snowflake/ml/monitoring/model_monitor_version.py,sha256=TlmDJZDE0lCVatRaBRgXIjzDF538nrMIc-zWj9MM_nk,46
 snowflake/ml/monitoring/shap.py,sha256=Dp9nYquPEZjxMTW62YYA9g9qUdmCEFxcSk7ejvOP7PE,3597
-snowflake/ml/monitoring/_client/model_monitor_sql_client.py,sha256=EH2jTz6ctNxfXbOMGVbQTAgikUig5YmvSsX93cd9ZF8,20194
+snowflake/ml/monitoring/_client/model_monitor_sql_client.py,sha256=Qr3L6bs84ID5_1TvY6wf5YK2kn3ZVZ-Havo242i3MiY,12710
 snowflake/ml/monitoring/_client/queries/record_count.ssql,sha256=Bd1uNMwhPKqPyrDd5ug8iY493t9KamJjrlo82OAfmjY,335
 snowflake/ml/monitoring/_client/queries/rmse.ssql,sha256=OEJiSStRz9-qKoZaFvmubtY_n0xMUjyVU2uiQHCp7KU,822
-snowflake/ml/monitoring/_manager/model_monitor_manager.py,sha256=gT3aYZsjD5wIRLdbe7fyyb5vICIxw9WWsK7H0hxbz9E,10314
-snowflake/ml/monitoring/entities/model_monitor_config.py,sha256=wvk0v9-VvhFAaNdpYXSqKdWj2Kx-KGjuWVkaCgL4MUc,825
-snowflake/ml/monitoring/entities/output_score_type.py,sha256=UJyS4z5hncRZ0agVNa6_X041RY9q3Us-6Bh3dPVAmEw,2982
+snowflake/ml/monitoring/_manager/model_monitor_manager.py,sha256=_-vxqnHqohTHTrwfURjPXijyAeh1mTRdHCG436GaBik,10314
+snowflake/ml/monitoring/entities/model_monitor_config.py,sha256=IxEiee1HfBXCQGzJOZbrDrvoV8J1tDNk43ygNuN00Io,1793
 snowflake/ml/registry/__init__.py,sha256=XdPQK9ejYkSJVrSQ7HD3jKQO0hKq2mC4bPCB6qrtH3U,76
-snowflake/ml/registry/registry.py,sha256=_G6Sm4Zi67iJJ3RUwz2XNYszPnrOtYF5bK8KeGtjubM,23793
+snowflake/ml/registry/registry.py,sha256=5aBedBH8NiFkJJe1Pnggsrjnn0ixdg1oqtUHWyz3wsE,23824
 snowflake/ml/registry/_manager/model_manager.py,sha256=gFr1EqaMR2Eb4erwVz7fi7xK1G1YsFXz1PF5GvOR0pg,12131
+snowflake/ml/utils/authentication.py,sha256=Wx1kVBZ9XBDuKkRHpPEB2pBxpiJepVLFAirDMx4m5Gk,2612
 snowflake/ml/utils/connection_params.py,sha256=JRpQppuWRk6bhdLzVDhMfz3Y6yInobFNLHmIBaXD7po,8005
 snowflake/ml/utils/sparse.py,sha256=XqDQkw39Ml6YIknswdkvFIwUwBk_GBXAbP8IACfPENg,3817
 snowflake/ml/utils/sql_client.py,sha256=z4Rhi7pQz3s9cyu_Uzfr3deCnrkCdFh9IYIvicsuwdc,692
-snowflake_ml_python-1.7.1.dist-info/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
-snowflake_ml_python-1.7.1.dist-info/METADATA,sha256=s1vUDI47E0APJr53Bs6qTDV-fWFk3gLkW9yImkzM960,65547
-snowflake_ml_python-1.7.1.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-snowflake_ml_python-1.7.1.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
-snowflake_ml_python-1.7.1.dist-info/RECORD,,
+snowflake_ml_python-1.7.2.dist-info/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
+snowflake_ml_python-1.7.2.dist-info/METADATA,sha256=GwZOHmNQAKaMDP3VeWIDWC-OMhPqldoJaYPrR-_iWGw,67429
+snowflake_ml_python-1.7.2.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+snowflake_ml_python-1.7.2.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
+snowflake_ml_python-1.7.2.dist-info/RECORD,,

{snowflake_ml_python-1.7.1.dist-info → snowflake_ml_python-1.7.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.3.0)
+Generator: setuptools (75.6.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

snowflake/ml/monitoring/entities/output_score_type.py

@@ -1,90 +0,0 @@
-from __future__ import annotations
-
-from enum import Enum
-from typing import List, Mapping
-
-from snowflake.ml._internal.utils import sql_identifier
-from snowflake.ml.model import type_hints
-from snowflake.snowpark import types
-
-# Accepted data types for each OutputScoreType.
-REGRESSION_DATA_TYPES = (
-    types.ByteType,
-    types.ShortType,
-    types.IntegerType,
-    types.LongType,
-    types.FloatType,
-    types.DoubleType,
-    types.DecimalType,
-)
-CLASSIFICATION_DATA_TYPES = (
-    types.ByteType,
-    types.ShortType,
-    types.IntegerType,
-    types.BooleanType,
-    types.BinaryType,
-)
-PROBITS_DATA_TYPES = (
-    types.ByteType,
-    types.ShortType,
-    types.IntegerType,
-    types.LongType,
-    types.FloatType,
-    types.DoubleType,
-    types.DecimalType,
-)
-
-
-# OutputScoreType enum
-class OutputScoreType(Enum):
-    UNKNOWN = "UNKNOWN"
-    REGRESSION = "REGRESSION"
-    CLASSIFICATION = "CLASSIFICATION"
-    PROBITS = "PROBITS"
-
-    @classmethod
-    def deduce_score_type(
-        cls,
-        table_schema: Mapping[str, types.DataType],
-        prediction_columns: List[sql_identifier.SqlIdentifier],
-        task: type_hints.Task,
-    ) -> OutputScoreType:
-        """Find the score type for monitoring given a table schema and the task.
-
-        Args:
-            table_schema: Dictionary of column names and types in the source table.
-            prediction_columns: List of prediction columns.
-            task: Enum value for the task of the model.
-
-        Returns:
-            Enum value for the score type, informing monitoring table set up.
-
-        Raises:
-            ValueError: If prediction type fails to align with task.
-        """
-        # Already validated we have just one prediction column type
-        prediction_column_type = {table_schema[column_name] for column_name in prediction_columns}.pop()
-
-        if task == type_hints.Task.TABULAR_REGRESSION:
-            if isinstance(prediction_column_type, REGRESSION_DATA_TYPES):
-                return OutputScoreType.REGRESSION
-            else:
-                raise ValueError(
-                    f"Expected prediction column type to be one of {REGRESSION_DATA_TYPES} "
-                    f"for REGRESSION task. Found: {prediction_column_type}."
-                )
-
-        elif task == type_hints.Task.TABULAR_BINARY_CLASSIFICATION:
-            if isinstance(prediction_column_type, CLASSIFICATION_DATA_TYPES):
-                return OutputScoreType.CLASSIFICATION
-            elif isinstance(prediction_column_type, PROBITS_DATA_TYPES):
-                return OutputScoreType.PROBITS
-            else:
-                raise ValueError(
-                    f"Expected prediction column type to be one of {CLASSIFICATION_DATA_TYPES} "
-                    f"or one of {PROBITS_DATA_TYPES} for CLASSIFICATION task. "
-                    f"Found: {prediction_column_type}."
-                )
-
-        else:
-            raise ValueError(f"Received unsupported task for model monitoring: {task}.")