snowflake-ml-python 1.8.3__py3-none-any.whl → 1.8.4__py3-none-any.whl

snowflake/ml/registry/registry.py

@@ -148,11 +148,11 @@ class Registry:
                 dependencies must be retrieved from Snowflake Anaconda Channel.
             artifact_repository_map: Specifies a mapping of package channels or platforms to custom artifact
                 repositories. Defaults to None. Currently, the mapping applies only to warehouse execution.
-                Note : This feature is currently in Private Preview; please contact your Snowflake account team
-                to enable it.
+                Note : This feature is currently in Public Preview.
                 Format: {channel_name: artifact_repository_name}, where:
-                   - channel_name: The name of the Conda package channel (e.g., 'condaforge') or 'pip' for pip packages.
-                   - artifact_repository_name: The name or URL of the repository to fetch packages from.
+                   - channel_name: Currently must be 'pip'.
+                   - artifact_repository_name: The identifier of the artifact repository to fetch packages from, e.g.
+                     `snowflake.snowpark.pypi_shared_repository`.
             resource_constraint: Mapping of resource constraint keys and values, e.g. {"architecture": "x86"}.
             target_platforms: List of target platforms to run the model. The only acceptable inputs are a combination of
                 {"WAREHOUSE", "SNOWPARK_CONTAINER_SERVICES"}. Defaults to None.
@@ -288,14 +288,15 @@ class Registry:
                 dependencies must be retrieved from Snowflake Anaconda Channel.
             artifact_repository_map: Specifies a mapping of package channels or platforms to custom artifact
                 repositories. Defaults to None. Currently, the mapping applies only to warehouse execution.
-                Note : This feature is currently in Private Preview; please contact your Snowflake account team to
-                enable it.
+                Note : This feature is currently in Public Preview.
                 Format: {channel_name: artifact_repository_name}, where:
-                   - channel_name: The name of the Conda package channel (e.g., 'condaforge') or 'pip' for pip packages.
-                   - artifact_repository_name: The name or URL of the repository to fetch packages from.
+                   - channel_name: Currently must be 'pip'.
+                   - artifact_repository_name: The identifier of the artifact repository to fetch packages from, e.g.
+                     `snowflake.snowpark.pypi_shared_repository`.
             resource_constraint: Mapping of resource constraint keys and values, e.g. {"architecture": "x86"}.
             target_platforms: List of target platforms to run the model. The only acceptable inputs are a combination of
-                {"WAREHOUSE", "SNOWPARK_CONTAINER_SERVICES"}. Defaults to None.
+                ["WAREHOUSE", "SNOWPARK_CONTAINER_SERVICES"]. Defaults to None. When None, the target platforms will be
+                both.
             python_version: Python version in which the model is run. Defaults to None.
             signatures: Model data signatures for inputs and outputs for various target methods. If it is None,
                 sample_input_data would be used to infer the signatures for those models that cannot automatically

snowflake/ml/version.py

@@ -1,2 +1,2 @@
 # This is parsed by regex in conda recipe meta file. Make sure not to break it.
-VERSION = "1.8.3"
+VERSION = "1.8.4"

{snowflake_ml_python-1.8.3.dist-info → snowflake_ml_python-1.8.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.8.3
+Version: 1.8.4
 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: 
@@ -236,7 +236,7 @@ License-File: LICENSE.txt
 Requires-Dist: absl-py<2,>=0.15
 Requires-Dist: anyio<5,>=3.5.0
 Requires-Dist: cachetools<6,>=3.1.1
-Requires-Dist: cloudpickle<3,>=2.0.0
+Requires-Dist: cloudpickle>=2.0.0
 Requires-Dist: cryptography
 Requires-Dist: fsspec[http]<2026,>=2024.6.1
 Requires-Dist: importlib_resources<7,>=6.1.1
@@ -250,27 +250,31 @@ Requires-Dist: pytimeparse<2,>=1.1.8
 Requires-Dist: pyyaml<7,>=6.0
 Requires-Dist: retrying<2,>=1.3.3
 Requires-Dist: s3fs<2026,>=2024.6.1
-Requires-Dist: scikit-learn<1.6,>=1.4
+Requires-Dist: scikit-learn<1.6
 Requires-Dist: scipy<2,>=1.9
-Requires-Dist: snowflake-connector-python[pandas]<4,>=3.12.0
+Requires-Dist: shap<1,>=0.46.0
+Requires-Dist: snowflake-connector-python[pandas]<4,>=3.14.0
 Requires-Dist: snowflake-snowpark-python!=1.26.0,<2,>=1.17.0
 Requires-Dist: snowflake.core<2,>=1.0.2
 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: altair<6,>=5; extra == "all"
 Requires-Dist: catboost<2,>=1.2.0; extra == "all"
 Requires-Dist: keras<4,>=2.0.0; extra == "all"
 Requires-Dist: lightgbm<5,>=4.1.0; extra == "all"
 Requires-Dist: mlflow<3,>=2.16.0; extra == "all"
 Requires-Dist: sentence-transformers<4,>=2.7.0; extra == "all"
 Requires-Dist: sentencepiece<0.2.0,>=0.1.95; extra == "all"
-Requires-Dist: shap<1,>=0.46.0; extra == "all"
+Requires-Dist: streamlit<2,>=1.44.0; extra == "all"
 Requires-Dist: tensorflow<3,>=2.17.0; extra == "all"
 Requires-Dist: tokenizers<1,>=0.15.1; extra == "all"
 Requires-Dist: torch<3,>=2.0.1; extra == "all"
 Requires-Dist: torchdata<1,>=0.4; extra == "all"
 Requires-Dist: transformers<5,>=4.39.3; extra == "all"
+Requires-Dist: xgboost<3,>=1.7.3; extra == "all"
+Provides-Extra: altair
+Requires-Dist: altair<6,>=5; extra == "altair"
 Provides-Extra: catboost
 Requires-Dist: catboost<2,>=1.2.0; extra == "catboost"
 Provides-Extra: keras
@@ -281,8 +285,8 @@ Provides-Extra: lightgbm
 Requires-Dist: lightgbm<5,>=4.1.0; extra == "lightgbm"
 Provides-Extra: mlflow
 Requires-Dist: mlflow<3,>=2.16.0; extra == "mlflow"
-Provides-Extra: shap
-Requires-Dist: shap<1,>=0.46.0; extra == "shap"
+Provides-Extra: streamlit
+Requires-Dist: streamlit<2,>=1.44.0; extra == "streamlit"
 Provides-Extra: tensorflow
 Requires-Dist: tensorflow<3,>=2.17.0; extra == "tensorflow"
 Provides-Extra: torch
@@ -294,6 +298,8 @@ Requires-Dist: sentencepiece<0.2.0,>=0.1.95; extra == "transformers"
 Requires-Dist: tokenizers<1,>=0.15.1; extra == "transformers"
 Requires-Dist: torch<3,>=2.0.1; extra == "transformers"
 Requires-Dist: transformers<5,>=4.39.3; extra == "transformers"
+Provides-Extra: xgboost
+Requires-Dist: xgboost<3,>=1.7.3; extra == "xgboost"
 Dynamic: license-file
 
 # Snowpark ML
@@ -404,6 +410,31 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
+## 1.8.4
+
+### Bug Fixes
+
+- Registry: Default `enable_explainability` to True when the model can be deployed to Warehouse.
+- Registry: Add `custom_model.partitioned_api` decorator and deprecate `partitioned_inference_api`.
+- Registry: Fixed a bug when logging pytroch and tensorflow models that caused
+  `UnboundLocalError: local variable 'multiple_inputs' referenced before assignment`.
+
+### Breaking change
+
+- ML Job: Updated property `id` to be fully qualified name; Introduced new property `name` to represent the ML Job name
+- ML Job: Modified `list_jobs()` to return ML Job `name` instead of `id`
+- Registry: Error in `log_model` if `enable_explainability` is True and model is only deployed to
+   Snowpark Container Services, instead of just user warning.
+
+### New Features
+
+- ML Job: Extend `@remote` function decorator, `submit_file()` and `submit_directory()` to accept `database` and
+  `schema` parameters
+- ML Job: Support querying by fully qualified name in `get_job()`
+- Explainability: Added visualization functions to `snowflake.ml.monitoring` to plot explanations in notebooks.
+- Explainability: Support explain for categorical transforms for sklearn pipeline
+- Support categorical type for `xgboost.DMatrix` inputs.
+
 ## 1.8.3
 
 ### Bug Fixes
@@ -417,6 +448,7 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
   as a list of strings
 - Registry: Support `ModelVersion.run_job` to run inference with a single-node Snowpark Container Services job.
 - DataConnector: Removed PrPr decorators
+- Registry: Default the target platform to warehouse when logging a partitioned model.
 
 ## 1.8.2