snowflake-ml-python 1.54.0__tar.gz → 2.1.0__tar.gz

{snowflake_ml_python-1.54.0 → snowflake_ml_python-2.1.0}/CHANGELOG.md

@@ -1,6 +1,111 @@
 # Release History
 
-## 1.54.0
+## 2.1.0
+
+### New Features
+
+* Experiment Tracking: Added `ExperimentTracking.get_metric_history`, which returns every logged step of a
+  metric instead of only the value at the highest step reported by `list_metrics`. Pass a metric name to scope
+  the result to one metric, or omit it to get every metric of the run. The result is a lazy Snowpark DataFrame
+  with `name`, `step`, `value`, and `timestamp` columns, so filtering and aggregation run in Snowflake.
+
+* Feature Store: `list_feature_views()` now surfaces the online feature table's setup readiness in the
+  `online_config` JSON as `setup_status`, `setup_error_msg` and `setup_time`, for both batch/streaming
+  and realtime feature views. The keys are present only when setup-readiness information is available
+  for the table, so test for key presence. `SETUP_NOTREADY` means setup has not concluded yet, not
+  that setup failed; `SETUP_FAILED` is the failure signal. Note that `SETUP_READY` is also reported
+  when nothing has ever been reported for the table, and in that case `setup_status` may later change
+  to `SETUP_FAILED`.
+
+### Bug Fixes
+
+* Registry: Fixed logging of MLflow models created with `mlflow.sklearn.save_model()`. The model is
+  now re-logged from the scikit-learn estimator using the serialization format it was saved with,
+  instead of from MLflow's PyFunc wrapper with the version-dependent default format.
+
+### Behavior Changes
+
+### Deprecations
+
+## 2.0.0 (2026-09-09)
+
+Release candidate for the `snowflake-ml-python` 2.0 major release. This is a breaking
+release that removes deprecated APIs and slims down the default install footprint.
+
+### New Features
+
+* Feature Store: added `alter_online_service(size=...)`, which changes the size of an existing Online Service. The
+  call returns as soon as the request is recorded; the change runs in the background and can take hours on a large
+  Online Service, which keeps serving online reads and stream ingestion throughout. Poll
+  `get_online_service_status()` until `status` is `RUNNING` at the requested size.
+
+* ML Jobs: `submit_file`, `submit_directory`, `submit_from_stage`, and `@remote` accept a `comment` kwarg, attached as
+  the comment on the job service. When omitted, the comment defaults to the `application` name configured on the
+  session, so jobs submitted by a tool or partner integration remain attributable after the fact.
+
+### Bug Fixes
+
+* Feature Store: `delete_feature_view` now drops the online feature table before the
+  offline dynamic table or view, so a dependent FeatureGroup (error 099940) does not
+  leave an orphaned offline object.
+
+### Behavior Changes
+
+* Registry: Models logged from a Snowflake Container Runtime notebook without an explicit
+  `target_platforms` now default to both Warehouse and Snowpark Container Services instead of
+  Snowpark Container Services only. Warehouse is no longer dropped from the default target
+  platforms.
+* Dependencies: `scikit-learn`, `xgboost`, and `shap` are no longer required dependencies of
+  `snowflake-ml-python`. They are now optional extras. Install them with
+  `pip install snowflake-ml-python[scikit-learn]`, `[xgboost]`, `[shap]`, or `[all]` when using
+  the corresponding functionality. Extras are independent: `[xgboost]` and `[lightgbm]` install
+  only those libraries, not `scikit-learn`. Native XGBoost or LightGBM models (for example
+  logging to the Model Registry) only need `[xgboost]` or `[lightgbm]`.
+  `snowflake.ml.modeling` estimators always require `scikit-learn`, so the XGBoost and LightGBM
+  wrappers need `pip install "snowflake-ml-python[xgboost,scikit-learn]"` or
+  `"snowflake-ml-python[lightgbm,scikit-learn]"`.
+* Feature Store: Added an optional `feature_store` extra. Install `snowflake-ml-python[feature_store]`
+  to pull in `httpx`, which is required for online feature serving reads. `httpx` is not installed by
+  the default `snowflake-ml-python` install. Conda users still need to install `httpx` separately
+  (for example `conda install httpx`) to use the online Feature Store.
+* Modeling: Importing a `snowflake.ml.modeling` subpackage (for example
+  `snowflake.ml.modeling.linear_model`, `snowflake.ml.modeling.xgboost`,
+  `snowflake.ml.modeling.lightgbm`, or `snowflake.ml.modeling.preprocessing`) without the optional
+  dependencies it needs now raises an actionable `ImportError` that names only the missing
+  package(s), instead of a confusing missing-attribute error.
+  `scikit-learn` and `xgboost` are attributed to the 2.0 dependency change; `lightgbm` was
+  already an optional extra before 2.0.
+
+### Deprecations
+
+* Modeling: The `snowflake.ml.modeling` estimators and preprocessing transformers are deprecated and
+  will be removed in a future release. Importing a `snowflake.ml.modeling` subpackage now emits a
+  `DeprecationWarning`. Train models with the native scikit-learn, XGBoost, or LightGBM estimators
+  and log them to the Snowflake Model Registry (`snowflake.ml.registry`) instead.
+
+### Breaking Changes
+
+* Generic: Require python >= 3.10. Python 3.9 is no longer supported.
+* Removed the deprecated `snowflake.ml.utils.connection_params.SnowflakeLoginOptions` public API
+  (deprecated since 1.8.5).
+* Registry: `ModelVersion.run_batch` now runs on `EXECUTE INFERENCE JOB SERVICE`. The single
+  `job_spec` argument is replaced by the `resources_spec`, `inference_spec`, and `image_build_spec`
+  blocks, and input may be supplied as either a DataFrame (`X`) or an existing stage path
+  (`input_stage_location`). `output_spec.stage_location` is treated as a base: results are written
+  under a per-job subdirectory, `<stage_location>/<job_name>/`. `output_spec.base_stage_location`,
+  `job_spec.job_name_prefix`, and `job_spec.block` no longer exist.
+* Registry: Removed the `snowflake.ml.model.batch` module. Use `snowflake.ml.model.batch_inference`,
+  which exports `BatchInferenceTask` alongside the input, output, resources, inference, and image
+  build spec classes.
+* Feature Store: Feature views created before 1.42.0 that use the `last_distinct_n` or
+  `first_distinct_n` aggregation function can no longer be read offline. `generate_dataset`,
+  `generate_training_set`, `retrieve_feature_values`, and `read_feature_view` now raise a
+  `ValueError` for such feature views. Re-create the feature view to restore offline reads.
+* Dependencies: `snowflake-snowpark-python` must now be `>=1.37.0`. Earlier releases import
+  `pkg_resources`, which `setuptools` removed in 82.0.0, so they fail to import in environments
+  with a current `setuptools`.
+
+## 1.54.0 (2026-08-31)
 
 ### New Features
 
@@ -27,6 +132,9 @@
 * Feature Store: Fixed FeatureStore.update_feature_view() failing with `SQL compilation error:
   invalid value 'ADAPTIVE' for property 'REFRESH_MODE'` when enabling online storage for an existing
   feature view.
+* Feature Store: online feature reads and stream ingestion no longer fail while the Online Service is
+  changing size (status `UPDATING_SIZE`). The service serves reads and writes throughout a size
+  change, which can take hours, so it is now treated as serviceable like `UPDATING` already was.
 
 ### Behavior Changes
 
@@ -61,10 +169,6 @@
 
 ### Bug Fixes
 
-* Feature Store: online feature reads and stream ingestion no longer fail while the Online Service is
-  changing size (status `UPDATING_SIZE`). The service serves reads and writes throughout a size
-  change, which can take hours, so it is now treated as serviceable like `UPDATING` already was.
-
 ### Behavior Changes
 
 * Feature Store: `register_feature_view` now records `FV_SOURCE_REFS` metadata for managed batch

{snowflake_ml_python-1.54.0 → snowflake_ml_python-2.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-ml-python
-Version: 1.54.0
+Version: 2.1.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: 
@@ -220,7 +220,6 @@ Classifier: Intended Audience :: Information Technology
 Classifier: Intended Audience :: System Administrators
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
@@ -230,14 +229,13 @@ Classifier: Topic :: Software Development :: Libraries
 Classifier: Topic :: Software Development :: Libraries :: Application Frameworks
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: Scientific/Engineering :: Information Analysis
-Requires-Python: <3.14,>=3.9
+Requires-Python: <3.14,>=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
 Requires-Dist: anyio<5,>=3.5.0
 Requires-Dist: cachetools>=3.1.1
 Requires-Dist: cloudpickle>=2.0.0
 Requires-Dist: fsspec[http]<2026,>=2024.6.1
-Requires-Dist: h2<5,>=4.3.0
 Requires-Dist: importlib_resources<7,>=6.1.1
 Requires-Dist: jinja2<4,>=3.1.0
 Requires-Dist: numpy<3,>=1.23
@@ -250,34 +248,39 @@ 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: scikit-learn<1.9
 Requires-Dist: scipy<2,>=1.9
-Requires-Dist: shap<1,>=0.46.0
 Requires-Dist: snowflake-connector-python[pandas]<5,>=3.17.3
-Requires-Dist: snowflake-snowpark-python!=1.26.0,<2,>=1.17.0
+Requires-Dist: snowflake-snowpark-python<2,>=1.37.0
 Requires-Dist: sqlparse<1,>=0.4
 Requires-Dist: tqdm<5
 Requires-Dist: typing-extensions<5,>=4.1.0
-Requires-Dist: xgboost<4
 Provides-Extra: all
 Requires-Dist: altair<7,>=5; extra == "all"
 Requires-Dist: catboost<2,>=1.2.0; extra == "all"
+Requires-Dist: h2<5,>=4.3.0; extra == "all"
+Requires-Dist: httpx<1,>=0.28.1; extra == "all"
 Requires-Dist: keras<4,>=2.0.0; extra == "all"
 Requires-Dist: lightgbm<5,>=4.1.0; extra == "all"
 Requires-Dist: mlflow<4,>=2.16.0; extra == "all"
 Requires-Dist: prophet<2,>=1.1.0; extra == "all"
+Requires-Dist: scikit-learn<1.9; extra == "all"
 Requires-Dist: sentence-transformers<6,>=2.7.0; extra == "all"
 Requires-Dist: sentencepiece<0.3,>=0.1.95; extra == "all"
+Requires-Dist: shap<1,>=0.46.0; extra == "all"
 Requires-Dist: streamlit<2,>=1.30.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!=4.51.3,<6,>=4.39.3; extra == "all"
+Requires-Dist: xgboost<4; extra == "all"
 Provides-Extra: altair
 Requires-Dist: altair<7,>=5; extra == "altair"
 Provides-Extra: catboost
 Requires-Dist: catboost<2,>=1.2.0; extra == "catboost"
+Provides-Extra: feature-store
+Requires-Dist: h2<5,>=4.3.0; extra == "feature-store"
+Requires-Dist: httpx<1,>=0.28.1; extra == "feature-store"
 Provides-Extra: keras
 Requires-Dist: keras<4,>=2.0.0; extra == "keras"
 Requires-Dist: tensorflow<3,>=2.17.0; extra == "keras"
@@ -296,6 +299,10 @@ Provides-Extra: mlflow
 Requires-Dist: mlflow<4,>=2.16.0; extra == "mlflow"
 Provides-Extra: prophet
 Requires-Dist: prophet<2,>=1.1.0; extra == "prophet"
+Provides-Extra: scikit-learn
+Requires-Dist: scikit-learn<1.9; extra == "scikit-learn"
+Provides-Extra: shap
+Requires-Dist: shap<1,>=0.46.0; extra == "shap"
 Provides-Extra: streamlit
 Requires-Dist: streamlit<2,>=1.30.0; extra == "streamlit"
 Provides-Extra: tensorflow
@@ -309,6 +316,8 @@ Requires-Dist: sentencepiece<0.3,>=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!=4.51.3,<6,>=4.39.3; extra == "transformers"
+Provides-Extra: xgboost
+Requires-Dist: xgboost<4; extra == "xgboost"
 Dynamic: license-file
 
 # Snowflake ML Python
@@ -370,7 +379,7 @@ In Snowflake Warehouse notebook environments, snowflake-ml-python can be install
 Follow the [installation instructions](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#installing-snowpark-ml)
 in the Snowflake documentation.
 
-Python versions 3.9 to 3.12 are supported. You can use [miniconda](https://docs.conda.io/en/latest/miniconda.html) or
+Python versions 3.10 to 3.13 are supported. You can use [miniconda](https://docs.conda.io/en/latest/miniconda.html) or
 [anaconda](https://www.anaconda.com/) to create a Conda environment (recommended),
 or [virtualenv](https://docs.python.org/3/tutorial/venv.html) to create a virtual environment.
 
@@ -416,7 +425,112 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 # Release History
 
-## 1.54.0
+## 2.1.0
+
+### New Features
+
+* Experiment Tracking: Added `ExperimentTracking.get_metric_history`, which returns every logged step of a
+  metric instead of only the value at the highest step reported by `list_metrics`. Pass a metric name to scope
+  the result to one metric, or omit it to get every metric of the run. The result is a lazy Snowpark DataFrame
+  with `name`, `step`, `value`, and `timestamp` columns, so filtering and aggregation run in Snowflake.
+
+* Feature Store: `list_feature_views()` now surfaces the online feature table's setup readiness in the
+  `online_config` JSON as `setup_status`, `setup_error_msg` and `setup_time`, for both batch/streaming
+  and realtime feature views. The keys are present only when setup-readiness information is available
+  for the table, so test for key presence. `SETUP_NOTREADY` means setup has not concluded yet, not
+  that setup failed; `SETUP_FAILED` is the failure signal. Note that `SETUP_READY` is also reported
+  when nothing has ever been reported for the table, and in that case `setup_status` may later change
+  to `SETUP_FAILED`.
+
+### Bug Fixes
+
+* Registry: Fixed logging of MLflow models created with `mlflow.sklearn.save_model()`. The model is
+  now re-logged from the scikit-learn estimator using the serialization format it was saved with,
+  instead of from MLflow's PyFunc wrapper with the version-dependent default format.
+
+### Behavior Changes
+
+### Deprecations
+
+## 2.0.0 (2026-09-09)
+
+Release candidate for the `snowflake-ml-python` 2.0 major release. This is a breaking
+release that removes deprecated APIs and slims down the default install footprint.
+
+### New Features
+
+* Feature Store: added `alter_online_service(size=...)`, which changes the size of an existing Online Service. The
+  call returns as soon as the request is recorded; the change runs in the background and can take hours on a large
+  Online Service, which keeps serving online reads and stream ingestion throughout. Poll
+  `get_online_service_status()` until `status` is `RUNNING` at the requested size.
+
+* ML Jobs: `submit_file`, `submit_directory`, `submit_from_stage`, and `@remote` accept a `comment` kwarg, attached as
+  the comment on the job service. When omitted, the comment defaults to the `application` name configured on the
+  session, so jobs submitted by a tool or partner integration remain attributable after the fact.
+
+### Bug Fixes
+
+* Feature Store: `delete_feature_view` now drops the online feature table before the
+  offline dynamic table or view, so a dependent FeatureGroup (error 099940) does not
+  leave an orphaned offline object.
+
+### Behavior Changes
+
+* Registry: Models logged from a Snowflake Container Runtime notebook without an explicit
+  `target_platforms` now default to both Warehouse and Snowpark Container Services instead of
+  Snowpark Container Services only. Warehouse is no longer dropped from the default target
+  platforms.
+* Dependencies: `scikit-learn`, `xgboost`, and `shap` are no longer required dependencies of
+  `snowflake-ml-python`. They are now optional extras. Install them with
+  `pip install snowflake-ml-python[scikit-learn]`, `[xgboost]`, `[shap]`, or `[all]` when using
+  the corresponding functionality. Extras are independent: `[xgboost]` and `[lightgbm]` install
+  only those libraries, not `scikit-learn`. Native XGBoost or LightGBM models (for example
+  logging to the Model Registry) only need `[xgboost]` or `[lightgbm]`.
+  `snowflake.ml.modeling` estimators always require `scikit-learn`, so the XGBoost and LightGBM
+  wrappers need `pip install "snowflake-ml-python[xgboost,scikit-learn]"` or
+  `"snowflake-ml-python[lightgbm,scikit-learn]"`.
+* Feature Store: Added an optional `feature_store` extra. Install `snowflake-ml-python[feature_store]`
+  to pull in `httpx`, which is required for online feature serving reads. `httpx` is not installed by
+  the default `snowflake-ml-python` install. Conda users still need to install `httpx` separately
+  (for example `conda install httpx`) to use the online Feature Store.
+* Modeling: Importing a `snowflake.ml.modeling` subpackage (for example
+  `snowflake.ml.modeling.linear_model`, `snowflake.ml.modeling.xgboost`,
+  `snowflake.ml.modeling.lightgbm`, or `snowflake.ml.modeling.preprocessing`) without the optional
+  dependencies it needs now raises an actionable `ImportError` that names only the missing
+  package(s), instead of a confusing missing-attribute error.
+  `scikit-learn` and `xgboost` are attributed to the 2.0 dependency change; `lightgbm` was
+  already an optional extra before 2.0.
+
+### Deprecations
+
+* Modeling: The `snowflake.ml.modeling` estimators and preprocessing transformers are deprecated and
+  will be removed in a future release. Importing a `snowflake.ml.modeling` subpackage now emits a
+  `DeprecationWarning`. Train models with the native scikit-learn, XGBoost, or LightGBM estimators
+  and log them to the Snowflake Model Registry (`snowflake.ml.registry`) instead.
+
+### Breaking Changes
+
+* Generic: Require python >= 3.10. Python 3.9 is no longer supported.
+* Removed the deprecated `snowflake.ml.utils.connection_params.SnowflakeLoginOptions` public API
+  (deprecated since 1.8.5).
+* Registry: `ModelVersion.run_batch` now runs on `EXECUTE INFERENCE JOB SERVICE`. The single
+  `job_spec` argument is replaced by the `resources_spec`, `inference_spec`, and `image_build_spec`
+  blocks, and input may be supplied as either a DataFrame (`X`) or an existing stage path
+  (`input_stage_location`). `output_spec.stage_location` is treated as a base: results are written
+  under a per-job subdirectory, `<stage_location>/<job_name>/`. `output_spec.base_stage_location`,
+  `job_spec.job_name_prefix`, and `job_spec.block` no longer exist.
+* Registry: Removed the `snowflake.ml.model.batch` module. Use `snowflake.ml.model.batch_inference`,
+  which exports `BatchInferenceTask` alongside the input, output, resources, inference, and image
+  build spec classes.
+* Feature Store: Feature views created before 1.42.0 that use the `last_distinct_n` or
+  `first_distinct_n` aggregation function can no longer be read offline. `generate_dataset`,
+  `generate_training_set`, `retrieve_feature_values`, and `read_feature_view` now raise a
+  `ValueError` for such feature views. Re-create the feature view to restore offline reads.
+* Dependencies: `snowflake-snowpark-python` must now be `>=1.37.0`. Earlier releases import
+  `pkg_resources`, which `setuptools` removed in 82.0.0, so they fail to import in environments
+  with a current `setuptools`.
+
+## 1.54.0 (2026-08-31)
 
 ### New Features
 
@@ -443,6 +557,9 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 * Feature Store: Fixed FeatureStore.update_feature_view() failing with `SQL compilation error:
   invalid value 'ADAPTIVE' for property 'REFRESH_MODE'` when enabling online storage for an existing
   feature view.
+* Feature Store: online feature reads and stream ingestion no longer fail while the Online Service is
+  changing size (status `UPDATING_SIZE`). The service serves reads and writes throughout a size
+  change, which can take hours, so it is now treated as serviceable like `UPDATING` already was.
 
 ### Behavior Changes
 
@@ -477,10 +594,6 @@ NOTE: Version 1.7.0 is used as example here. Please choose the the latest versio
 
 ### Bug Fixes
 
-* Feature Store: online feature reads and stream ingestion no longer fail while the Online Service is
-  changing size (status `UPDATING_SIZE`). The service serves reads and writes throughout a size
-  change, which can take hours, so it is now treated as serviceable like `UPDATING` already was.
-
 ### Behavior Changes
 
 * Feature Store: `register_feature_view` now records `FV_SOURCE_REFS` metadata for managed batch

{snowflake_ml_python-1.54.0 → snowflake_ml_python-2.1.0}/README.md

@@ -57,7 +57,7 @@ In Snowflake Warehouse notebook environments, snowflake-ml-python can be install
 Follow the [installation instructions](https://docs.snowflake.com/en/developer-guide/snowpark-ml/index#installing-snowpark-ml)
 in the Snowflake documentation.
 
-Python versions 3.9 to 3.12 are supported. You can use [miniconda](https://docs.conda.io/en/latest/miniconda.html) or
+Python versions 3.10 to 3.13 are supported. You can use [miniconda](https://docs.conda.io/en/latest/miniconda.html) or
 [anaconda](https://www.anaconda.com/) to create a Conda environment (recommended),
 or [virtualenv](https://docs.python.org/3/tutorial/venv.html) to create a virtual environment.
 

{snowflake_ml_python-1.54.0 → snowflake_ml_python-2.1.0}/pyproject.toml

@@ -5,10 +5,10 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "snowflake-ml-python"
 description = "The machine learning client library that is used for interacting with Snowflake to build machine learning solutions."
-classifiers = [ "Development Status :: 5 - Production/Stable", "Environment :: Console", "Environment :: Other Environment", "Intended Audience :: Developers", "Intended Audience :: Education", "Intended Audience :: Information Technology", "Intended Audience :: System Administrators", "License :: OSI Approved :: Apache Software License", "Operating System :: OS Independent", "Programming Language :: Python :: 3.9", "Programming Language :: Python :: 3.10", "Programming Language :: Python :: 3.11", "Programming Language :: Python :: 3.12", "Topic :: Database", "Topic :: Software Development", "Topic :: Software Development :: Libraries", "Topic :: Software Development :: Libraries :: Application Frameworks", "Topic :: Software Development :: Libraries :: Python Modules", "Topic :: Scientific/Engineering :: Information Analysis",]
-requires-python = ">=3.9, <3.14"
+classifiers = [ "Development Status :: 5 - Production/Stable", "Environment :: Console", "Environment :: Other Environment", "Intended Audience :: Developers", "Intended Audience :: Education", "Intended Audience :: Information Technology", "Intended Audience :: System Administrators", "License :: OSI Approved :: Apache Software License", "Operating System :: OS Independent", "Programming Language :: Python :: 3.10", "Programming Language :: Python :: 3.11", "Programming Language :: Python :: 3.12", "Topic :: Database", "Topic :: Software Development", "Topic :: Software Development :: Libraries", "Topic :: Software Development :: Libraries :: Application Frameworks", "Topic :: Software Development :: Libraries :: Python Modules", "Topic :: Scientific/Engineering :: Information Analysis",]
+requires-python = ">=3.10, <3.14"
 dynamic = [ "version", "readme",]
-dependencies = [ "anyio>=3.5.0,<5", "cachetools>=3.1.1", "cloudpickle>=2.0.0", "fsspec[http]>=2024.6.1,<2026", "h2>=4.3.0,<5", "importlib_resources>=6.1.1, <7", "jinja2>=3.1.0,<4", "numpy>=1.23,<3", "packaging>=20.9,<26", "pandas>=2.1.4,<3", "platformdirs<5", "pyarrow", "pydantic>=2.8.2, <3", "pyjwt>=2.0.0, <3", "pytimeparse>=1.1.8,<2", "pyyaml>=6.0,<7", "retrying>=1.3.3,<2", "scikit-learn<1.9", "scipy>=1.9,<2", "shap>=0.46.0,<1", "snowflake-connector-python[pandas]>=3.17.3,<5", "snowflake-snowpark-python>=1.17.0,<2,!=1.26.0", "sqlparse>=0.4,<1", "tqdm<5", "typing-extensions>=4.1.0,<5", "xgboost<4",]
+dependencies = [ "anyio>=3.5.0,<5", "cachetools>=3.1.1", "cloudpickle>=2.0.0", "fsspec[http]>=2024.6.1,<2026", "importlib_resources>=6.1.1, <7", "jinja2>=3.1.0,<4", "numpy>=1.23,<3", "packaging>=20.9,<26", "pandas>=2.1.4,<3", "platformdirs<5", "pyarrow", "pydantic>=2.8.2, <3", "pyjwt>=2.0.0, <3", "pytimeparse>=1.1.8,<2", "pyyaml>=6.0,<7", "retrying>=1.3.3,<2", "scipy>=1.9,<2", "snowflake-connector-python[pandas]>=3.17.3,<5", "snowflake-snowpark-python>=1.37.0,<2", "sqlparse>=0.4,<1", "tqdm<5", "typing-extensions>=4.1.0,<5",]
 [[project.authors]]
 name = "Snowflake, Inc"
 email = "support@snowflake.com"
@@ -24,18 +24,22 @@ Issues = "https://github.com/snowflakedb/snowflake-ml-python/issues"
 Changelog = "https://github.com/snowflakedb/snowflake-ml-python/blob/master/CHANGELOG.md"
 
 [project.optional-dependencies]
-all = [ "altair>=5,<7", "catboost>=1.2.0, <2", "keras>=2.0.0,<4", "lightgbm>=4.1.0, <5", "mlflow>=2.16.0, <4", "prophet>=1.1.0, <2", "sentence-transformers>=2.7.0,<6", "sentencepiece>=0.1.95,<0.3", "streamlit>=1.30.0,<2", "tensorflow>=2.17.0,<3", "tokenizers>=0.15.1,<1", "torch>=2.0.1,<3", "torchdata>=0.4,<1", "transformers>=4.39.3,!=4.51.3,<6",]
+all = [ "altair>=5,<7", "catboost>=1.2.0, <2", "h2>=4.3.0,<5", "httpx>=0.28.1,<1", "keras>=2.0.0,<4", "lightgbm>=4.1.0, <5", "mlflow>=2.16.0, <4", "prophet>=1.1.0, <2", "scikit-learn<1.9", "sentence-transformers>=2.7.0,<6", "sentencepiece>=0.1.95,<0.3", "shap>=0.46.0,<1", "streamlit>=1.30.0,<2", "tensorflow>=2.17.0,<3", "tokenizers>=0.15.1,<1", "torch>=2.0.1,<3", "torchdata>=0.4,<1", "transformers>=4.39.3,!=4.51.3,<6", "xgboost<4",]
 altair = [ "altair>=5,<7",]
 catboost = [ "catboost>=1.2.0, <2",]
+feature_store = [ "h2>=4.3.0,<5", "httpx>=0.28.1,<1",]
 keras = [ "keras>=2.0.0,<4", "tensorflow>=2.17.0,<3", "torch>=2.0.1,<3",]
 lightgbm = [ "lightgbm>=4.1.0, <5",]
 llm = [ "mlflow>=2.16.0, <4", "sentence-transformers>=2.7.0,<6", "sentencepiece>=0.1.95,<0.3", "tokenizers>=0.15.1,<1", "torch>=2.0.1,<3", "torchdata>=0.4,<1", "transformers>=4.39.3,!=4.51.3,<6",]
 mlflow = [ "mlflow>=2.16.0, <4",]
 prophet = [ "prophet>=1.1.0, <2",]
+scikit-learn = [ "scikit-learn<1.9",]
+shap = [ "shap>=0.46.0,<1",]
 streamlit = [ "streamlit>=1.30.0,<2",]
 tensorflow = [ "tensorflow>=2.17.0,<3",]
 torch = [ "torch>=2.0.1,<3", "torchdata>=0.4,<1",]
 transformers = [ "sentence-transformers>=2.7.0,<6", "sentencepiece>=0.1.95,<0.3", "tokenizers>=0.15.1,<1", "torch>=2.0.1,<3", "transformers>=4.39.3,!=4.51.3,<6",]
+xgboost = [ "xgboost<4",]
 
 [tool.setuptools.package-data]
 "*" = [ "*",]

{snowflake_ml_python-1.54.0 → snowflake_ml_python-2.1.0}/snowflake/ml/_internal/env_utils.py

@@ -8,7 +8,7 @@ import textwrap
 import warnings
 from enum import Enum
 from importlib import metadata as importlib_metadata
-from typing import Any, DefaultDict, Optional
+from typing import Any, DefaultDict
 
 import yaml
 from packaging import requirements, specifiers, version
@@ -376,53 +376,13 @@ def get_package_spec_with_supported_ops_only(req: requirements.Requirement) -> r
         A requirements.Requirement object with supported ops only
     """
 
-    if req.name == "numpy":
-        import numpy as np
-
-        package_specifiers = get_numpy_specifiers(req, version.Version(np.__version__).major)
-    else:
-        package_specifiers = [spec for spec in req.specifier if spec.operator in _SUPPORTED_PACKAGE_SPEC_OPS]
+    package_specifiers = [spec for spec in req.specifier if spec.operator in _SUPPORTED_PACKAGE_SPEC_OPS]
 
     new_req = copy.deepcopy(req)
     new_req.specifier = specifiers.SpecifierSet(specifiers=",".join([str(spec) for spec in package_specifiers]))
     return new_req
 
 
-def get_numpy_specifiers(
-    req: requirements.Requirement,
-    client_numpy_major_version: int,
-) -> list[specifiers.Specifier]:
-    """Get the package spec with supported ops only including ==, >=, <=, > and < based on the client numpy
-    major version.
-
-    Args:
-        req: A requirements.Requirement object showing the requirement.
-        client_numpy_major_version: The major version of numpy to be used.
-
-    Returns:
-        A list of specifiers with supported ops only
-    """
-    req_specifiers = []
-    for org_spec in req.specifier:
-        # check specifier that provides upper bound
-        if org_spec.operator in ["<", "<="]:
-            client_version = version.Version(str(client_numpy_major_version))
-            org_spec_version = version.Version(org_spec.version)
-            # check if the client's numpy major version is less than the specifier's upper bound
-            # if so, pin to max possible client major version
-            if client_version.major < org_spec_version.major:
-                modified_spec = specifiers.Specifier(f"<{client_version.major + 1}")
-                req_specifiers.append(modified_spec)
-            else:
-                # use the original specifier
-                req_specifiers.append(org_spec)
-        else:
-            # use the original specifier
-            req_specifiers.append(org_spec)
-
-    return req_specifiers
-
-
 def _relax_specifier_set(
     specifier_set: specifiers.SpecifierSet, strategy: relax_version_strategy.RelaxVersionStrategy
 ) -> specifiers.SpecifierSet:
@@ -493,7 +453,7 @@ def get_matched_package_versions_in_information_schema(
     session: session.Session,
     reqs: list[requirements.Requirement],
     python_version: str,
-    statement_params: Optional[dict[str, Any]] = None,
+    statement_params: dict[str, Any] | None = None,
 ) -> dict[str, list[version.Version]]:
     """Look up the information_schema table to check if a package with the specified specifier exists in the Snowflake
     Conda channel. Note that this is not the source of truth due to the potential delay caused by a package that might
@@ -567,7 +527,7 @@ def save_conda_env_file(
     path: pathlib.Path,
     conda_chan_deps: DefaultDict[str, list[requirements.Requirement]],
     python_version: str,
-    cuda_version: Optional[str] = None,
+    cuda_version: str | None = None,
     default_channel_override: str = SNOWFLAKE_CONDA_CHANNEL_URL,
 ) -> None:
     """Generate conda.yml file given a dict of dependencies after validation.
@@ -688,7 +648,7 @@ def _has_torch_dependency(pip_deps: list[requirements.Requirement]) -> bool:
 def save_requirements_file(
     path: pathlib.Path,
     pip_deps: list[requirements.Requirement],
-    cuda_version: Optional[str] = None,
+    cuda_version: str | None = None,
 ) -> list[str]:
     """Generate Python requirements.txt file in the given directory path.
 
@@ -727,9 +687,9 @@ def load_conda_env_file(
     path: pathlib.Path,
 ) -> tuple[
     DefaultDict[str, list[requirements.Requirement]],
-    Optional[list[requirements.Requirement]],
-    Optional[str],
-    Optional[str],
+    list[requirements.Requirement] | None,
+    str | None,
+    str | None,
 ]:
     """Read conda.yml file to get a dict of dependencies after validation.
     The channels part of conda.yml file will be processed with following rules:
@@ -830,7 +790,7 @@ def load_requirements_file(path: pathlib.Path) -> list[requirements.Requirement]
 PYTHON_VERSION_PATTERN = re.compile(r"python(?:(?P<op>=|==|>|<|>=|<=|~=|===)(?P<ver>\d(?:\.\d+)+))?(\.*)?")
 
 
-def parse_python_version_string(dep: str) -> Optional[str]:
+def parse_python_version_string(dep: str) -> str | None:
     if dep.startswith("python"):
         m = PYTHON_VERSION_PATTERN.match(dep)
         if m is None:
@@ -849,7 +809,7 @@ def parse_python_version_string(dep: str) -> Optional[str]:
 
 def _find_conda_dep_spec(
     conda_chan_deps: DefaultDict[str, list[requirements.Requirement]], pkg_name: str
-) -> Optional[tuple[str, requirements.Requirement]]:
+) -> tuple[str, requirements.Requirement] | None:
     for channel in conda_chan_deps:
         spec = next(filter(lambda req: req.name == pkg_name, conda_chan_deps[channel]), None)
         if spec:
@@ -857,7 +817,7 @@ def _find_conda_dep_spec(
     return None
 
 
-def _find_pip_req_spec(pip_reqs: list[requirements.Requirement], pkg_name: str) -> Optional[requirements.Requirement]:
+def _find_pip_req_spec(pip_reqs: list[requirements.Requirement], pkg_name: str) -> requirements.Requirement | None:
     spec = next(filter(lambda req: req.name == pkg_name, pip_reqs), None)
     return spec
 
@@ -866,9 +826,9 @@ def find_dep_spec(
     conda_chan_deps: DefaultDict[str, list[requirements.Requirement]],
     pip_reqs: list[requirements.Requirement],
     conda_pkg_name: str,
-    pip_pkg_name: Optional[str] = None,
+    pip_pkg_name: str | None = None,
     remove_spec: bool = False,
-) -> Optional[requirements.Requirement]:
+) -> requirements.Requirement | None:
     if pip_pkg_name is None:
         pip_pkg_name = conda_pkg_name
     spec_conda = _find_conda_dep_spec(conda_chan_deps, conda_pkg_name)