snowflake-ml-python 1.3.1__py3-none-any.whl → 1.4.0__py3-none-any.whl

snowflake/ml/modeling/lightgbm/lgbm_classifier.py

@@ -16,6 +16,7 @@ from numpy import typing as npt
 
 
 import numpy
+import sklearn
 import lightgbm
 from sklearn.utils.metaestimators import available_if
 
@@ -160,7 +161,7 @@ class LGBMClassifier(BaseTransformer):
         self.set_sample_weight_col(sample_weight_col)
         self._use_external_memory_version = False
         self._batch_size = -1        
-        deps: Set[str] = set([f'numpy=={np.__version__}', f'lightgbm=={lightgbm.__version__}', f'cloudpickle=={cp.__version__}'])
+        deps: Set[str] = set([f'numpy=={np.__version__}', f'lightgbm=={lightgbm.__version__}', f'cloudpickle=={cp.__version__}', f'scikit-learn=={sklearn.__version__}'])
         
         self._deps = list(deps)
         
@@ -841,7 +842,7 @@ class LGBMClassifier(BaseTransformer):
             transform_kwargs = dict(
                 session=dataset._session,
                 dependencies=["snowflake-snowpark-python"] + self._deps,
-                score_sproc_imports=['lightgbm'],
+                score_sproc_imports=['lightgbm', 'sklearn'],
             )
         elif isinstance(dataset, pd.DataFrame):
             # pandas_handler.score() does not require any extra kwargs.

snowflake/ml/modeling/lightgbm/lgbm_regressor.py

@@ -16,6 +16,7 @@ from numpy import typing as npt
 
 
 import numpy
+import sklearn
 import lightgbm
 from sklearn.utils.metaestimators import available_if
 
@@ -160,7 +161,7 @@ class LGBMRegressor(BaseTransformer):
         self.set_sample_weight_col(sample_weight_col)
         self._use_external_memory_version = False
         self._batch_size = -1        
-        deps: Set[str] = set([f'numpy=={np.__version__}', f'lightgbm=={lightgbm.__version__}', f'cloudpickle=={cp.__version__}'])
+        deps: Set[str] = set([f'numpy=={np.__version__}', f'lightgbm=={lightgbm.__version__}', f'cloudpickle=={cp.__version__}', f'scikit-learn=={sklearn.__version__}'])
         
         self._deps = list(deps)
         
@@ -837,7 +838,7 @@ class LGBMRegressor(BaseTransformer):
             transform_kwargs = dict(
                 session=dataset._session,
                 dependencies=["snowflake-snowpark-python"] + self._deps,
-                score_sproc_imports=['lightgbm'],
+                score_sproc_imports=['lightgbm', 'sklearn'],
             )
         elif isinstance(dataset, pd.DataFrame):
             # pandas_handler.score() does not require any extra kwargs.

snowflake/ml/registry/_manager/model_manager.py

@@ -4,6 +4,7 @@ from typing import Any, Dict, List, Optional
 import pandas as pd
 from absl.logging import logging
 
+from snowflake.ml._internal.human_readable_id import hrid_generator
 from snowflake.ml._internal.utils import sql_identifier
 from snowflake.ml.model import model_signature, type_hints as model_types
 from snowflake.ml.model._client.model import model_impl, model_version_impl
@@ -27,13 +28,14 @@ class ModelManager:
         self._model_ops = model_ops.ModelOperator(
             session, database_name=self._database_name, schema_name=self._schema_name
         )
+        self._hrid_generator = hrid_generator.HRID16()
 
     def log_model(
         self,
         model: model_types.SupportedModelType,
         *,
         model_name: str,
-        version_name: str,
+        version_name: Optional[str] = None,
         comment: Optional[str] = None,
         metrics: Optional[Dict[str, Any]] = None,
         conda_dependencies: Optional[List[str]] = None,
@@ -48,6 +50,8 @@ class ModelManager:
     ) -> model_version_impl.ModelVersion:
         model_name_id = sql_identifier.SqlIdentifier(model_name)
 
+        if not version_name:
+            version_name = self._hrid_generator.generate()[1]
         version_name_id = sql_identifier.SqlIdentifier(version_name)
 
         if self._model_ops.validate_existence(

snowflake/ml/registry/model_registry.py

@@ -103,7 +103,10 @@ def _create_registry_database(
 
 
 def _create_registry_schema(
-    session: snowpark.Session, database_name: str, schema_name: str, statement_params: Dict[str, Any]
+    session: snowpark.Session,
+    database_name: str,
+    schema_name: str,
+    statement_params: Dict[str, Any],
 ) -> None:
     """Private helper to create the model registry schema.
 
@@ -161,7 +164,11 @@ def _create_registry_views(
 
     # Create a view on active permanent deployments.
     _create_active_permanent_deployment_view(
-        session, fully_qualified_schema_name, registry_table_name, deployment_table_name, statement_params
+        session,
+        fully_qualified_schema_name,
+        registry_table_name,
+        deployment_table_name,
+        statement_params,
     )
 
     # Create views on most recent metadata items.
@@ -437,10 +444,12 @@ fully integrated into the new registry.
         # Could do a multi-table insert here with some pros and cons:
         # [PRO] Atomic insert across multiple tables.
         # [CON] Code logic becomes messy depending on which fields are set.
-        # [CON] Harder to re-use existing methods like set_model_name.
+        # [CON] Harder to reuse existing methods like set_model_name.
         # Context: https://docs.snowflake.com/en/sql-reference/sql/insert-multi-table.html
         return table_manager.insert_table_entry(
-            self._session, table=self._fully_qualified_registry_table_name(), columns=properties
+            self._session,
+            table=self._fully_qualified_registry_table_name(),
+            columns=properties,
         )
 
     def _insert_metadata_entry(self, *, id: str, attribute: str, value: Any, operation: str) -> List[snowpark.Row]:
@@ -471,7 +480,9 @@ fully integrated into the new registry.
         columns["VALUE"] = value
 
         return table_manager.insert_table_entry(
-            self._session, table=self._fully_qualified_metadata_table_name(), columns=columns
+            self._session,
+            table=self._fully_qualified_metadata_table_name(),
+            columns=columns,
         )
 
     def _insert_deployment_entry(
@@ -484,7 +495,10 @@ fully integrated into the new registry.
         signature: Dict[str, Any],
         target_method: str,
         options: Optional[
-            Union[model_types.WarehouseDeployOptions, model_types.SnowparkContainerServiceDeployOptions]
+            Union[
+                model_types.WarehouseDeployOptions,
+                model_types.SnowparkContainerServiceDeployOptions,
+            ]
         ] = None,
     ) -> List[snowpark.Row]:
         """Insert a new row into the model deployment table.
@@ -521,7 +535,9 @@ fully integrated into the new registry.
         columns["OPTIONS"] = options
 
         return table_manager.insert_table_entry(
-            self._session, table=self._fully_qualified_deployment_table_name(), columns=columns
+            self._session,
+            table=self._fully_qualified_deployment_table_name(),
+            columns=columns,
         )
 
     def _prepare_deployment_stage(self) -> str:
@@ -596,7 +612,11 @@ fully integrated into the new registry.
         return identifier.get_schema_level_object_identifier(db, schema, stage)
 
     def _list_selected_models(
-        self, *, id: Optional[str] = None, model_name: Optional[str] = None, model_version: Optional[str] = None
+        self,
+        *,
+        id: Optional[str] = None,
+        model_name: Optional[str] = None,
+        model_version: Optional[str] = None,
     ) -> snowpark.DataFrame:
         """Retrieve the Snowpark dataframe of models matching the specified ID or (name and version).
 
@@ -724,7 +744,12 @@ fully integrated into the new registry.
         assert id is not None
 
         try:
-            self._insert_metadata_entry(id=id, attribute=attribute, value={attribute: value}, operation=operation)
+            self._insert_metadata_entry(
+                id=id,
+                attribute=attribute,
+                value={attribute: value},
+                operation=operation,
+            )
         except connector.DataError:
             raise connector.DataError(f"Setting {attribute} for mode id {id} failed.")
 
@@ -760,7 +785,10 @@ fully integrated into the new registry.
         return str(result)
 
     def _get_model_path(
-        self, id: Optional[str] = None, model_name: Optional[str] = None, model_version: Optional[str] = None
+        self,
+        id: Optional[str] = None,
+        model_name: Optional[str] = None,
+        model_version: Optional[str] = None,
     ) -> str:
         """Get the stage path for the model with the given (model name + model version) or `id` from the registry.
 
@@ -889,10 +917,17 @@ fully integrated into the new registry.
                 value=new_model,
             )
             if description:
-                self.set_model_description(model_name=model_name, model_version=model_version, description=description)
+                self.set_model_description(
+                    model_name=model_name,
+                    model_version=model_version,
+                    description=description,
+                )
             if tags:
                 self._set_metadata_attribute(
-                    _METADATA_ATTRIBUTE_TAGS, value=tags, model_name=model_name, model_version=model_version
+                    _METADATA_ATTRIBUTE_TAGS,
+                    value=tags,
+                    model_name=model_name,
+                    model_version=model_version,
                 )
         else:
             raise connector.DatabaseError("Failed to insert the model properties to the registry table.")
@@ -961,7 +996,10 @@ fully integrated into the new registry.
         model_tags = self.get_tags(model_name=model_name, model_version=model_version)
         model_tags[tag_name] = tag_value
         self._set_metadata_attribute(
-            _METADATA_ATTRIBUTE_TAGS, model_tags, model_name=model_name, model_version=model_version
+            _METADATA_ATTRIBUTE_TAGS,
+            model_tags,
+            model_name=model_name,
+            model_version=model_version,
         )
 
     @telemetry.send_api_usage_telemetry(
@@ -991,7 +1029,10 @@ fully integrated into the new registry.
             )
 
         self._set_metadata_attribute(
-            _METADATA_ATTRIBUTE_TAGS, model_tags, model_name=model_name, model_version=model_version
+            _METADATA_ATTRIBUTE_TAGS,
+            model_tags,
+            model_name=model_name,
+            model_version=model_version,
         )
 
     @telemetry.send_api_usage_telemetry(
@@ -1089,7 +1130,9 @@ fully integrated into the new registry.
             Description of the model or None.
         """
         result = self._get_metadata_attribute(
-            _METADATA_ATTRIBUTE_DESCRIPTION, model_name=model_name, model_version=model_version
+            _METADATA_ATTRIBUTE_DESCRIPTION,
+            model_name=model_name,
+            model_version=model_version,
         )
         return None if result is None else json.loads(result)
 
@@ -1112,7 +1155,10 @@ fully integrated into the new registry.
             description: Desired new model description.
         """
         self._set_metadata_attribute(
-            _METADATA_ATTRIBUTE_DESCRIPTION, description, model_name=model_name, model_version=model_version
+            _METADATA_ATTRIBUTE_DESCRIPTION,
+            description,
+            model_name=model_name,
+            model_version=model_version,
         )
 
     @telemetry.send_api_usage_telemetry(
@@ -1165,7 +1211,10 @@ fully integrated into the new registry.
             snowpark.DataFrame with the history of the model.
         """
         id = self._get_model_id(model_name=model_name, model_version=model_version)
-        return cast(snowpark.DataFrame, self.get_history().filter(snowpark.Column("MODEL_ID") == id))
+        return cast(
+            snowpark.DataFrame,
+            self.get_history().filter(snowpark.Column("MODEL_ID") == id),
+        )
 
     @telemetry.send_api_usage_telemetry(
         project=_TELEMETRY_PROJECT,
@@ -1194,7 +1243,10 @@ fully integrated into the new registry.
         model_metrics = self.get_metrics(model_name=model_name, model_version=model_version)
         model_metrics[metric_name] = metric_value
         self._set_metadata_attribute(
-            _METADATA_ATTRIBUTE_METRICS, model_metrics, model_name=model_name, model_version=model_version
+            _METADATA_ATTRIBUTE_METRICS,
+            model_metrics,
+            model_name=model_name,
+            model_version=model_version,
         )
 
     @telemetry.send_api_usage_telemetry(
@@ -1230,7 +1282,10 @@ fully integrated into the new registry.
             )
 
         self._set_metadata_attribute(
-            _METADATA_ATTRIBUTE_METRICS, model_metrics, model_name=model_name, model_version=model_version
+            _METADATA_ATTRIBUTE_METRICS,
+            model_metrics,
+            model_name=model_name,
+            model_version=model_version,
         )
 
     @telemetry.send_api_usage_telemetry(
@@ -1290,7 +1345,9 @@ fully integrated into the new registry.
         # Snowpark snowpark.dataframe returns dictionary objects as strings. We need to convert it back to a dictionary
         # here.
         result = self._get_metadata_attribute(
-            _METADATA_ATTRIBUTE_METRICS, model_name=model_name, model_version=model_version
+            _METADATA_ATTRIBUTE_METRICS,
+            model_name=model_name,
+            model_version=model_version,
         )
 
         if result:
@@ -1507,7 +1564,10 @@ fully integrated into the new registry.
         permanent: bool = False,
         platform: deploy_platforms.TargetPlatform = deploy_platforms.TargetPlatform.WAREHOUSE,
         options: Optional[
-            Union[model_types.WarehouseDeployOptions, model_types.SnowparkContainerServiceDeployOptions]
+            Union[
+                model_types.WarehouseDeployOptions,
+                model_types.SnowparkContainerServiceDeployOptions,
+            ]
         ] = None,
     ) -> model_types.Deployment:
         """Deploy the model with the given deployment name.
@@ -1772,7 +1832,9 @@ fully integrated into the new registry.
 
         """
         deployment = self._get_deployment(
-            model_name=model_name, model_version=model_version, deployment_name=deployment_name
+            model_name=model_name,
+            model_version=model_version,
+            deployment_name=deployment_name,
         )
 
         # TODO(SNOW-759526): The following sequence should be a transaction.
@@ -1845,7 +1907,8 @@ fully integrated into the new registry.
 
         # Step 1/3: Delete the registry entry.
         query_result_checker.SqlResultValidator(
-            self._session, f"DELETE FROM {self._fully_qualified_registry_table_name()} WHERE ID='{id}'"
+            self._session,
+            f"DELETE FROM {self._fully_qualified_registry_table_name()} WHERE ID='{id}'",
         ).deletion_success(expected_num_rows=1).validate()
 
         # Step 2/3: Delete the artifact (if desired).
@@ -1966,7 +2029,11 @@ class ModelReference:
 
             def build_method(m: Callable[..., Any]) -> Callable[..., Any]:
                 return lambda self, *args, **kwargs: m(
-                    self._registry, self._model_name, self._model_version, *args, **kwargs
+                    self._registry,
+                    self._model_name,
+                    self._model_version,
+                    *args,
+                    **kwargs,
                 )
 
             method = build_method(m=obj)
@@ -2027,7 +2094,10 @@ class ModelReference:
 
         if di:
             return model_api.predict(
-                session=self._registry._session, deployment=di, X=data, statement_params=statement_params
+                session=self._registry._session,
+                deployment=di,
+                X=data,
+                statement_params=statement_params,
             )
 
         # Mypy enforce to refer to the registry for calling the function
@@ -2059,7 +2129,10 @@ class ModelReference:
             options=options,
         )
         return model_api.predict(
-            session=self._registry._session, deployment=di, X=data, statement_params=statement_params
+            session=self._registry._session,
+            deployment=di,
+            X=data,
+            statement_params=statement_params,
         )
 
 

snowflake/ml/registry/registry.py

@@ -77,7 +77,7 @@ class Registry:
         model: model_types.SupportedModelType,
         *,
         model_name: str,
-        version_name: str,
+        version_name: Optional[str] = None,
         comment: Optional[str] = None,
         metrics: Optional[Dict[str, Any]] = None,
         conda_dependencies: Optional[List[str]] = None,
@@ -98,6 +98,7 @@ class Registry:
                 Sentence Transformers, Peft-finetuned LLM, or Custom Model.
             model_name: Name to identify the model.
             version_name: Version identifier for the model. Combination of model_name and version_name must be unique.
+                If not specified, a random name will be generated.
             comment: Comment associated with the model version. Defaults to None.
             metrics: A JSON serializable dictionary containing metrics linked to the model version. Defaults to None.
             signatures: Model data signatures for inputs and outputs for various target methods. If it is None,

snowflake/ml/version.py

@@ -1 +1 @@
-VERSION="1.3.1"
+VERSION="1.4.0"

{snowflake_ml_python-1.3.1.dist-info → snowflake_ml_python-1.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: snowflake-ml-python
-Version: 1.3.1
+Version: 1.4.0
 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: 
@@ -370,13 +370,41 @@ be compatibility issues. Server-side functionality that `snowflake-ml-python` de
 
 # Release History
 
-## 1.3.1
+## 1.4.0
+
+### Bug Fixes
+
+- Registry: Fix a bug when multiple models are being called from the same query, models other than the first one will
+  have incorrect result. This fix only works for newly logged model.
+- Modeling: When registering a model, only method(s) that is mentioned in `save_model` would be added to model signature
+in SnowML models.
+- Modeling: Fix a bug that when n_jobs is not 1, model cannot execute methods such as
+predict, predict_log_proba, and other batch inference methods. The n_jobs would automatically
+set to 1 because vectorized udf currently doesn't support joblib parallel backend.
+- Modeling: Fix a bug that batch inference methods cannot infer the datatype when the first row of data contains NULL.
+- Modeling: Matches Distributed HPO output column names with the snowflake identifier.
+- Modeling: Relax package versions for all Distributed HPO methods if the installed version
+  is not available in the Snowflake conda channel
+- Modeling: Add sklearn as required dependency for LightGBM package.
+
+### Behavior Changes
+
+- Registry: `apply` method is no longer by default logged when logging a xgboost model. If that is required, it could
+  be specified manually when logging the model by `log_model(..., options={"target_methods": ["apply", ...]})`.
+
+### New Features
+
+- Registry: Add support for `sentence-transformers` model (`sentence_transformers.SentenceTransformer`).
+- Registry: Now version name is no longer required when logging a model. If not provided, a random human readable ID
+  will be generated.
+
+## 1.3.1 (2024-03-21)
 
 ### New Features
 
 - FileSet: `snowflake.ml.fileset.sfcfs.SFFileSystem` can now be used in UDFs and stored procedures.
 
-## 1.3.0
+## 1.3.0 (2024-03-12)
 
 ### Bug Fixes
 

{snowflake_ml_python-1.3.1.dist-info → snowflake_ml_python-1.4.0.dist-info}/RECORD

@@ -5,7 +5,7 @@ snowflake/cortex/_sentiment.py,sha256=7X_a8qJNFFgn-Y1tjwMDkyNJHz5yYl0PvnezVCc4Ts
 snowflake/cortex/_summarize.py,sha256=DJRxUrPrTVmtQNgus0ZPF1z8nPmn4Rs5oL3U25CfXxQ,1075
 snowflake/cortex/_translate.py,sha256=JPMIXxHTgJPfJqT5Hw_WtYM6FZ8NuQufZ4XR-M8wnyo,1420
 snowflake/cortex/_util.py,sha256=0xDaDSctenhuj59atZenZp5q9zuhji0WQ77KPjqqNoc,1557
-snowflake/ml/version.py,sha256=TPzBy2mBfZlYdDDim4fBFJhFtPDF4tRr8Te3p6dI8rs,16
+snowflake/ml/version.py,sha256=X9n40H72i_qUrqdGO45MIHs_PUCysRlD1xrJRZ97KiU,16
 snowflake/ml/_internal/env.py,sha256=kCrJTRnqQ97VGUVI1cWUPD8HuBWeL5vOOtwUR0NB9Mg,161
 snowflake/ml/_internal/env_utils.py,sha256=nkBk8bDDKi5zIK9ZD8hBlKd3krccNZ4XC2pt6bgb4L4,25797
 snowflake/ml/_internal/file_utils.py,sha256=OyXHv-UcItiip1YgLnab6etonUQkYuyDtmplZA0CaoU,13622
@@ -23,7 +23,11 @@ snowflake/ml/_internal/exceptions/exceptions.py,sha256=ub0fthrNTVoKhpj1pXnKRfO1G
 snowflake/ml/_internal/exceptions/fileset_error_messages.py,sha256=dqPpRu0cKyQA_0gahvbizgQBTwNhnwveN286JrJLvi8,419
 snowflake/ml/_internal/exceptions/fileset_errors.py,sha256=ZJfkpeDgRIw3qA876fk9FIzxIrm-yZ8I9RXUbzaeM84,1040
 snowflake/ml/_internal/exceptions/modeling_error_messages.py,sha256=q1Nh7KvnUebdKCwwAPmotdAVS578CgAXcfDOfKoweVw,665
-snowflake/ml/_internal/utils/formatting.py,sha256=udoXzwbgeZ6NTUeU7ywgSA4pASv3xtxm-IslW1l6ZqM,3677
+snowflake/ml/_internal/human_readable_id/adjectives.txt,sha256=5o4MbVeHoELAqyLpyuKleOKR47jPjC_nKoziOIZMwT0,804
+snowflake/ml/_internal/human_readable_id/animals.txt,sha256=GDLzMwzxiL07PhIMxw4t89bhYqqg0bQfPiuQT8VNeME,837
+snowflake/ml/_internal/human_readable_id/hrid_generator.py,sha256=LYWB86qZgsVBvnc6Q5VjfDOmnGSQU3cTRKfId_nJSPY,1341
+snowflake/ml/_internal/human_readable_id/hrid_generator_base.py,sha256=D1yoVG1vmAFUhWQ5xCRRU6HCCBPbXHpOXagFd0jK0O8,4519
+snowflake/ml/_internal/utils/formatting.py,sha256=PswZ6Xas7sx3Ok1MBLoH2o7nfXOxaJqpUPg_UqXrQb8,3676
 snowflake/ml/_internal/utils/identifier.py,sha256=_NAW00FGtQsQESxF2b30_T4kkmzQITsdfykvJ2PqPUo,10870
 snowflake/ml/_internal/utils/import_utils.py,sha256=eexwIe7auT17s4aVxAns7se0_K15rcq3O17MkIvDpPI,2068
 snowflake/ml/_internal/utils/log_stream_processor.py,sha256=pBf8ycEamhHjEzUT55Rx_tFqSkYRpD5Dt71Mx9ZdaS8,1001
@@ -43,7 +47,7 @@ snowflake/ml/_internal/utils/uri.py,sha256=pvskcWoeS0M66DaU2XlJzK9wce55z4J5dn5kT
 snowflake/ml/dataset/dataset.py,sha256=OG_RonPgj86mRKRgN-xhJV0uZfa78ohVBpxsoYYnceY,6078
 snowflake/ml/feature_store/__init__.py,sha256=dYtqk_GD_hAAZjGfH1maWlZQ30h4hu_KGaf-_y9_AD8,298
 snowflake/ml/feature_store/entity.py,sha256=dCpzLC3jrt5wDHqFYJXbAYkMiZ0zEmiVDMGkks6MXkA,3378
-snowflake/ml/feature_store/feature_store.py,sha256=s5TavQVlSnQo4n4N6aupnB4NufQ6EvJBpj9XtWU1yN0,73363
+snowflake/ml/feature_store/feature_store.py,sha256=aNcbXHwpq9qja3785kpvgvahGtooY2KGjqspLM26W1o,69115
 snowflake/ml/feature_store/feature_view.py,sha256=APSn-xqm1Yv_iIKCckPdsvAqFb7D0-3BUW6URjSNut8,17806
 snowflake/ml/fileset/fileset.py,sha256=QRhxLeKf1QBqvXO4RyyRd1c8TixhYpHuBEII8Qi3C_M,26201
 snowflake/ml/fileset/parquet_parser.py,sha256=sjyRB59cGBzSzvbcYLvu_ApMPtrR-zwZsQkxekMR4FA,6884
@@ -53,12 +57,12 @@ snowflake/ml/fileset/tf_dataset.py,sha256=K8jafWBsyRaIYEmxaYAYNDj3dLApK82cg0Mlx5
 snowflake/ml/fileset/torch_datapipe.py,sha256=O2irHckqLzPDnXemEbAEjc3ZCVnLufPdPbt9WKYiBp0,2386
 snowflake/ml/model/__init__.py,sha256=fk8OMvOyrSIkAhX0EcrgBBvdz1VGRsdMmfYFV2GCf14,367
 snowflake/ml/model/_api.py,sha256=Y3r-Rm1-TJ0rnuydcWs6ENGdNp86T57PbmCWJlB0o0U,21595
-snowflake/ml/model/custom_model.py,sha256=x1RczFD4cwlHwnQmRan5M6gN-71LNWXuiEk7nMici8Y,8185
+snowflake/ml/model/custom_model.py,sha256=xvu7WZ1YmOdvuPePyAj6qMwKq-HNeVV9bNfkOT09CRI,8267
 snowflake/ml/model/deploy_platforms.py,sha256=r6cS3gTNWG9i4P00fHehY6Q8eBiNva6501OTyp_E5m0,144
 snowflake/ml/model/model_signature.py,sha256=UQSGieGJcnmC02V4feCYMdhMXnGoOUa9KBuDrbeivBM,29342
 snowflake/ml/model/type_hints.py,sha256=qe9U01Br4zYN0Uo0Pm7OC8eyjIuAoVwzweSvEe9SMzQ,12195
 snowflake/ml/model/_client/model/model_impl.py,sha256=QmTJr1JLdqBHWrFFpR2xARfbx0INYPzbfKWJn--3yX4,12525
-snowflake/ml/model/_client/model/model_version_impl.py,sha256=KZNN_Jq2x69XwELEiVaAXkAGTMS1ZmjCFko8Ale9rBo,13333
+snowflake/ml/model/_client/model/model_version_impl.py,sha256=A9d4ipgiymX35Hsk7j4GkO908u0aVUAf95kUWybTT9M,13548
 snowflake/ml/model/_client/ops/metadata_ops.py,sha256=XFNolmueu0nC3nAjb2Lj3v1NffDAhAq0JWMek9JVO38,4094
 snowflake/ml/model/_client/ops/model_ops.py,sha256=cL791mSAr4fJvPco6PtMdpwqicHhSTc8nsn4jdcEuEA,17767
 snowflake/ml/model/_client/sql/model.py,sha256=diuyGfFtLu1Z9yBThP-SjGOG9Zy4gflRKh6JoyUBDHk,4525
@@ -69,12 +73,12 @@ snowflake/ml/model/_deploy_client/image_builds/base_image_builder.py,sha256=clCa
 snowflake/ml/model/_deploy_client/image_builds/client_image_builder.py,sha256=G74D9lV2B3d544YzFN-YrjPkaST7tbQeh-rM17dtoJc,10681
 snowflake/ml/model/_deploy_client/image_builds/docker_context.py,sha256=7uhAJsHsk7LbiZv_w3xOCE2O88rTUVnS3_B6OAz-JG4,6129
 snowflake/ml/model/_deploy_client/image_builds/gunicorn_run.sh,sha256=1pntXgqFthW4gdomqlyWx9CJF-Wqv8VMoLkgSiTHEJ0,1578
-snowflake/ml/model/_deploy_client/image_builds/server_image_builder.py,sha256=dvmK6ILhGc3z-oJDHDCPHxtIV8hfDpshRtawHZ7Wm-A,9986
+snowflake/ml/model/_deploy_client/image_builds/server_image_builder.py,sha256=HnTaj0v27R9PCRuXpcP1nWv5tGBsXGSq6Xwep1m0bb0,9947
 snowflake/ml/model/_deploy_client/image_builds/inference_server/main.py,sha256=Ltk7KrYsp-nrghMhbMWKqi3snU8inbqmKLHFFyBCeBY,11148
 snowflake/ml/model/_deploy_client/image_builds/templates/dockerfile_template,sha256=WAqYQaaY5AFywg9yNLKRw350c2fpM4vxgdjYJ50VJJA,1752
 snowflake/ml/model/_deploy_client/image_builds/templates/image_build_job_spec_template,sha256=g8mEvpJmwQ9OnAkZomeErPQ6h4OJ5NdtRCoylyIp7f4,1225
 snowflake/ml/model/_deploy_client/image_builds/templates/kaniko_shell_script_template,sha256=nEK7fqo_XHJEVKLNe--EkES4oiDm7M5E9CacxGItFU0,3633
-snowflake/ml/model/_deploy_client/snowservice/deploy.py,sha256=yOnYfKNxv3cFoi6w65i4GFYB8ul1YnMQdLymO6skqAU,28961
+snowflake/ml/model/_deploy_client/snowservice/deploy.py,sha256=U0axqxy9YdJTsGz0bXSRSM2f7nziRnB83mvK6Rz9tlI,29141
 snowflake/ml/model/_deploy_client/snowservice/deploy_options.py,sha256=X4ncWgcgS9DKaNDiauOR9aVC6D27yb3DNouXDEHEjMQ,5989
 snowflake/ml/model/_deploy_client/snowservice/instance_types.py,sha256=YHI5D7UXNlEbV_Bzk0Nq6nrzfv2VUJfxwchLe7hY-lA,232
 snowflake/ml/model/_deploy_client/snowservice/templates/service_spec_template,sha256=hZX8XYPAlEU2R6JhZLj46js91g7XSfe2pysflCYH4HM,734
@@ -83,12 +87,12 @@ snowflake/ml/model/_deploy_client/utils/constants.py,sha256=ysEBrEs0sBCGHnk9uBX-
 snowflake/ml/model/_deploy_client/utils/snowservice_client.py,sha256=R_ilt8SGwQR6qh_roaUvst0YrnjbJbAyxYIPn4efo4E,13284
 snowflake/ml/model/_deploy_client/warehouse/deploy.py,sha256=yZR9M76oh6JbPQJHb6t3wGO3wuD04w0zLEXiEyZW_tg,8358
 snowflake/ml/model/_deploy_client/warehouse/infer_template.py,sha256=1THMd6JX1nW-OozECyxXbn9HJXDgNBUIdhfC9ODPDWY,3011
-snowflake/ml/model/_model_composer/model_composer.py,sha256=KqZvrpxOzgHtxxCcLhBwCTXsmorGItLwagZb7rDY9Qk,6331
+snowflake/ml/model/_model_composer/model_composer.py,sha256=ShoSp74xImfdXuIMTVJKt09sIBS8uxz-0rCbYBxLX9o,6337
 snowflake/ml/model/_model_composer/model_manifest/model_manifest.py,sha256=wdMTFH8St31mr88Fj8lQLTj_gvskHQu8fQOxAPQoXuQ,6677
 snowflake/ml/model/_model_composer/model_manifest/model_manifest_schema.py,sha256=R4oX50Rlpr0C6zTYJRPuuZqImzYcBSTCQJfuSGutULI,2029
 snowflake/ml/model/_model_composer/model_method/function_generator.py,sha256=2B-fykyanYlGWA4Ie2nOwXx2N5D2qZEvTbbPuSSreeI,1837
 snowflake/ml/model/_model_composer/model_method/infer_function.py_template,sha256=QpQXAIKDs9cotLOL0JdI6xLet1QJU7KtaF7O10nDQcs,2291
-snowflake/ml/model/_model_composer/model_method/infer_table_function.py_template,sha256=8G2bHRCw1SB00HOHSkGh0SpUaPK7r7z5FGBmsfgWCRg,2164
+snowflake/ml/model/_model_composer/model_method/infer_table_function.py_template,sha256=gex5if17PZ6t6fPcr2i_LO_3IRY03Ykcv_XAyKJt8pg,2170
 snowflake/ml/model/_model_composer/model_method/model_method.py,sha256=Lk32tVHAN_oMRuKx_9hiFKuk7gqCDcJe-D0fN56BzvM,6693
 snowflake/ml/model/_model_composer/model_runtime/_runtime_requirements.py,sha256=z3V7mRgdP-TYpZSX7TrW2k_4hNQ3ZsR4YO4ZQ0YSm8s,248
 snowflake/ml/model/_model_composer/model_runtime/model_runtime.py,sha256=y6gZURScuGFZK6-n_YEdzDIzJHCiHXctKuSGv_ObRwc,4307
@@ -104,10 +108,10 @@ snowflake/ml/model/_packager/model_handlers/mlflow.py,sha256=Itw1fPiBdU2euOmjLU3
 snowflake/ml/model/_packager/model_handlers/pytorch.py,sha256=dSxKO530_IlF1OK3t9_UYpVntdPiszKy-x_7XGk0bzQ,8033
 snowflake/ml/model/_packager/model_handlers/sentence_transformers.py,sha256=JRPargMNEJaDFQIpzqEVvOml62G_UVVvJdqBH8Lhu_Y,9051
 snowflake/ml/model/_packager/model_handlers/sklearn.py,sha256=bb-3AkK5T5HlFLSzviGKKRjhVcGvKIClDU7OP1OsNHg,8065
-snowflake/ml/model/_packager/model_handlers/snowmlmodel.py,sha256=0Ea4Jf5gpbDWt9p6IWtsDRkohQUZMLf4UHc0ewaTmcE,7262
+snowflake/ml/model/_packager/model_handlers/snowmlmodel.py,sha256=le4Y_dbiPlcjhiFpK1shla3pVgQ5UASdx2g7a70tYYY,7967
 snowflake/ml/model/_packager/model_handlers/tensorflow.py,sha256=ujBcbJ1-Ymv7ZeLfuxuDBe7QZ7KNU7x1p2k6OM_yi-0,8179
 snowflake/ml/model/_packager/model_handlers/torchscript.py,sha256=8s8sMWQ9ydJpK1Nk2uPQ-FVeB-xclfX5qzRDr9G1bdk,8104
-snowflake/ml/model/_packager/model_handlers/xgboost.py,sha256=ZiHJ9nHwqFRpye6MZP-9wT6XlYZ4QDhh7WFkmTy0G0s,8883
+snowflake/ml/model/_packager/model_handlers/xgboost.py,sha256=x5bXz5DRzb3O7DMDOF535LBPGnydCa78JHP_7-vsnjY,8874
 snowflake/ml/model/_packager/model_handlers_migrator/base_migrator.py,sha256=BZo14UrywGZM1kTqzN4VFQcYjl7dggDp1U90ZBCMuOg,1409
 snowflake/ml/model/_packager/model_meta/_core_requirements.py,sha256=BE-T6xd48OmUIthNAapbI6w7cmUsJwd32I7c1slaXpE,274
 snowflake/ml/model/_packager/model_meta/_packaging_requirements.py,sha256=TfJNtrfyZoNiJZYFfmTbmiWMlXKM-QxkOBIJVFvPit0,44
@@ -130,7 +134,7 @@ snowflake/ml/model/models/huggingface_pipeline.py,sha256=62GpPZxBheqCnFNxNOggiDE
 snowflake/ml/model/models/llm.py,sha256=ofrdHH4LQEQmnxYAGwmHV2sWLPenf0WcgBLg9MPwSmY,3616
 snowflake/ml/modeling/_internal/constants.py,sha256=xI4ofa3ATQ2UszRPpkfUAxghV_gXmvxleqOew4UI1PM,45
 snowflake/ml/modeling/_internal/estimator_utils.py,sha256=Szhpip5g7ddmT1-nfRg8WFRRCBx9QIjsSW9ey7jkTLo,8855
-snowflake/ml/modeling/_internal/model_specifications.py,sha256=-0PWh4cy-XjbejGb00RiFTnBSWiYMTNFQntXTMADgko,4725
+snowflake/ml/modeling/_internal/model_specifications.py,sha256=nAqgw7i1LcYMKRQq9mg2I50Kl0tsayh2_do5UMDXdT0,4801
 snowflake/ml/modeling/_internal/model_trainer.py,sha256=AlnTRnIowaF39Qjy2Zv4U3JsMydzCxfcBB2pgLIzNpk,694
 snowflake/ml/modeling/_internal/model_trainer_builder.py,sha256=0zazMgVNmBly7jKLGEwwjirb6VUsmA5bnplCzWxfTP8,7269
 snowflake/ml/modeling/_internal/model_transformer_builder.py,sha256=Y6Y8XSr7X7xAy1FvjPuHTb9Opy7tnGoCuOUBc5WEBJ4,3364
@@ -139,8 +143,8 @@ snowflake/ml/modeling/_internal/local_implementations/pandas_handlers.py,sha256=
 snowflake/ml/modeling/_internal/local_implementations/pandas_trainer.py,sha256=QuXUeeFzktfxStkXFlFSzB7QAuaTG2mPQJVBlRkb0WI,3169
 snowflake/ml/modeling/_internal/ml_runtime_implementations/ml_runtime_handlers.py,sha256=y9PZ3xgPGDHPBcNHY0f2Fk0nMZMRsPcLWy2cIDTALT4,4850
 snowflake/ml/modeling/_internal/ml_runtime_implementations/ml_runtime_trainer.py,sha256=lM1vYwpJ1jgTh8vnuyMp4tnFibM6UFf50W1IpPWwUWE,2535
-snowflake/ml/modeling/_internal/snowpark_implementations/distributed_hpo_trainer.py,sha256=cXJ9ixStjzyMRMr4vEldYT8YzcmHjTqilYE_FphXuX4,29384
-snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_handlers.py,sha256=cxW8dW39fJQwt-y9OvysjYXW0nfkMJO7hwxE7lI338c,13405
+snowflake/ml/modeling/_internal/snowpark_implementations/distributed_hpo_trainer.py,sha256=1kfwnUzaBFcxrN_YjWdlbT6ZR_vqcBjUwscwDzTsQyQ,54431
+snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_handlers.py,sha256=qtizLi4azYYCPL0BzuxohK81BpxRPidQQGhwVgp2bAQ,13590
 snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_trainer.py,sha256=w3zCrv-TwDB1o9eFppMaiXWmMeEPz_EAn_vl_2_6GL8,21699
 snowflake/ml/modeling/_internal/snowpark_implementations/xgboost_external_memory_trainer.py,sha256=VBYWGTy6ajQ-u2aiEvVU6NnKobEqJyz65oaHJS-ZjBs,17208
 snowflake/ml/modeling/calibration/__init__.py,sha256=rY5qSOkHj59bHiTV6LhBiEhUA0StoCb0ACNR2vkV4v0,297
@@ -212,7 +216,7 @@ snowflake/ml/modeling/feature_selection/select_percentile.py,sha256=10-R2ijOsTVR
 snowflake/ml/modeling/feature_selection/sequential_feature_selector.py,sha256=kRZ4HPQgzawll6ZGZ76ZseVA3qhLYtNRFmq4d_9u0gc,46442
 snowflake/ml/modeling/feature_selection/variance_threshold.py,sha256=8i6_ALx-0XY2FxRGnBeOY2Jd29PYz6ALFDAouC6ysoQ,43402
 snowflake/ml/modeling/framework/_utils.py,sha256=85q83_QVwAQmnzMYefSE5FaxiGMYnOpRatyvdpemU6w,9974
-snowflake/ml/modeling/framework/base.py,sha256=TlAfli0RzDYr6SWxVfFcfkK6T9Up4XXZ0nnAYj9tMFg,29962
+snowflake/ml/modeling/framework/base.py,sha256=PteCPFStrGyeH1HP3oQLkY7AaNoWj8abHYwZuLoHfso,30157
 snowflake/ml/modeling/gaussian_process/__init__.py,sha256=rY5qSOkHj59bHiTV6LhBiEhUA0StoCb0ACNR2vkV4v0,297
 snowflake/ml/modeling/gaussian_process/gaussian_process_classifier.py,sha256=igmWhHpuU27OtYDAHbduJHNltCTKwakCu5ch3Q0brew,49376
 snowflake/ml/modeling/gaussian_process/gaussian_process_regressor.py,sha256=VXIWTMMM80zqzBBdwQLeEAdzf67jZ0pXhJ50O_do2UQ,48441
@@ -220,7 +224,7 @@ snowflake/ml/modeling/impute/__init__.py,sha256=dYtqk_GD_hAAZjGfH1maWlZQ30h4hu_K
 snowflake/ml/modeling/impute/iterative_imputer.py,sha256=jF_H8WvECHavwzaOKYqsR0ijM3VnyWzBC3Gl70Qd4zc,49939
 snowflake/ml/modeling/impute/knn_imputer.py,sha256=49U_EkD-kQ6uSIxC8BOKqc9IxS1IHuvW1gfOeGeEFCA,45696
 snowflake/ml/modeling/impute/missing_indicator.py,sha256=sdwF5ip-u7g3ioeK7ImdEmvCZE-D6xigAbiXP4kkduc,44524
-snowflake/ml/modeling/impute/simple_imputer.py,sha256=eC9CRhHMmsylm7oi5LGiukRlP6HIN6r3IP1DVpYrwmg,19258
+snowflake/ml/modeling/impute/simple_imputer.py,sha256=awM33HugS5jGs3JXud1U8eEMm2VLdIAf7z_eVXAzKD0,18499
 snowflake/ml/modeling/kernel_approximation/__init__.py,sha256=rY5qSOkHj59bHiTV6LhBiEhUA0StoCb0ACNR2vkV4v0,297
 snowflake/ml/modeling/kernel_approximation/additive_chi2_sampler.py,sha256=McvxnMNfyv3O2fNXsIMmGFFWfbKnEulj11Z0D4X8bDo,43515
 snowflake/ml/modeling/kernel_approximation/nystroem.py,sha256=fMANo2btM_MaJRID8RrF7Ni66uwcsvmyUlOXPEgo_4w,45319
@@ -230,8 +234,8 @@ snowflake/ml/modeling/kernel_approximation/skewed_chi2_sampler.py,sha256=0kB3UCP
 snowflake/ml/modeling/kernel_ridge/__init__.py,sha256=rY5qSOkHj59bHiTV6LhBiEhUA0StoCb0ACNR2vkV4v0,297
 snowflake/ml/modeling/kernel_ridge/kernel_ridge.py,sha256=Hm9RoSJqOvd6xQrNByvKU3NAjGktUF__Txsj5HrO66s,45755
 snowflake/ml/modeling/lightgbm/__init__.py,sha256=rY5qSOkHj59bHiTV6LhBiEhUA0StoCb0ACNR2vkV4v0,297
-snowflake/ml/modeling/lightgbm/lgbm_classifier.py,sha256=7iSIXWRBMmB8rzDtiMXzm43adHvtZAmBK4wv49fnljs,45257
-snowflake/ml/modeling/lightgbm/lgbm_regressor.py,sha256=eXxl0o7WYUAq9ToqOT5OKZg1YBCVvQ5_trlLgzVZusc,44760
+snowflake/ml/modeling/lightgbm/lgbm_classifier.py,sha256=tQWAMmWECo7i7q6L-RhzJZPTJ-EW90_U7VM0w7JCk5g,45323
+snowflake/ml/modeling/lightgbm/lgbm_regressor.py,sha256=llkP12nu04eOcC8DfeKyYx2YRLspxGXRP75V47YajSY,44826
 snowflake/ml/modeling/linear_model/__init__.py,sha256=rY5qSOkHj59bHiTV6LhBiEhUA0StoCb0ACNR2vkV4v0,297
 snowflake/ml/modeling/linear_model/ard_regression.py,sha256=BE0Mp7qtXFltlRrh25dytZla50nuyLeIvcE19wDsxDU,45700
 snowflake/ml/modeling/linear_model/bayesian_ridge.py,sha256=qU5_D_0H34X14gZa4EAdlxluPXqQIe5tRPPRctiZzOY,46116
@@ -354,13 +358,13 @@ snowflake/ml/registry/_schema.py,sha256=GOA427_mVKkq9RWRENHuqDimRS0SmmP4EWThNCu1
 snowflake/ml/registry/_schema_upgrade_plans.py,sha256=LxZNXYGjGG-NmB7w7_SxgaJpZuXUO66XVMuh04oL6SI,4209
 snowflake/ml/registry/_schema_version_manager.py,sha256=-9wGH-7ELSZxp7-fW7hXTMqkJSIebXdSpwwgzdvnoYs,6922
 snowflake/ml/registry/artifact.py,sha256=9JDcr4aaR0d4cp4YSRnGMFRIdu-k0tABbs6jDH4VDGQ,1263
-snowflake/ml/registry/model_registry.py,sha256=a3PqbATHNOG3oO3rY9zuoWMHd5jKKTl_or-HLU1PcgY,90041
-snowflake/ml/registry/registry.py,sha256=Hx-pjrnkwYSAftDaSLIXSgoivkXzXDrQxf-4-nzZ13E,10855
-snowflake/ml/registry/_manager/model_manager.py,sha256=67QHJi_ufloYmZikBALD_3MEdudeJDFtO0grRIimMkI,5578
+snowflake/ml/registry/model_registry.py,sha256=MgI4Dj9kvxfNd3kQ3tWY6ygmxUd6kzb430-GKkn4BA0,91007
+snowflake/ml/registry/registry.py,sha256=LIwExLFPMOvbJbB7nRToDkMk93wl1ZMhGiN1Mo5HRGk,10939
+snowflake/ml/registry/_manager/model_manager.py,sha256=LYX_nS_egwum7F_LCbz_a3hibIHOTDK8LO1DPOWxPrE,5809
 snowflake/ml/utils/connection_params.py,sha256=JRpQppuWRk6bhdLzVDhMfz3Y6yInobFNLHmIBaXD7po,8005
 snowflake/ml/utils/sparse.py,sha256=XqDQkw39Ml6YIknswdkvFIwUwBk_GBXAbP8IACfPENg,3817
-snowflake_ml_python-1.3.1.dist-info/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
-snowflake_ml_python-1.3.1.dist-info/METADATA,sha256=-h4LtktBD-3xkHOsGOgt4842Fz5l8gx7pAZL3U60rjw,45077
-snowflake_ml_python-1.3.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-snowflake_ml_python-1.3.1.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
-snowflake_ml_python-1.3.1.dist-info/RECORD,,
+snowflake_ml_python-1.4.0.dist-info/LICENSE.txt,sha256=PdEp56Av5m3_kl21iFkVTX_EbHJKFGEdmYeIO1pL_Yk,11365
+snowflake_ml_python-1.4.0.dist-info/METADATA,sha256=g26kO8pTVbH7coUpl7H3P4ceLOBSnilZNkT6U5ZTxGA,46650
+snowflake_ml_python-1.4.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+snowflake_ml_python-1.4.0.dist-info/top_level.txt,sha256=TY0gFSHKDdZy3THb0FGomyikWQasEGldIR1O0HGOHVw,10
+snowflake_ml_python-1.4.0.dist-info/RECORD,,