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

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,
@@ -93,11 +93,12 @@ class Registry:
         Log a model with various parameters and metadata.
 
         Args:
-            model: Model object of supported types such as Scikit-learn, XGBoost, Snowpark ML,
+            model: Model object of supported types such as Scikit-learn, XGBoost, LightGBM, Snowpark ML,
                 PyTorch, TorchScript, Tensorflow, Tensorflow Keras, MLFlow, HuggingFace Pipeline,
                 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.1"

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: snowflake-ml-python
-Version: 1.3.1
+Version: 1.4.1
 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,12 +236,13 @@ 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: catboost <1.3,>=1.2.0
 Requires-Dist: cloudpickle >=2.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 <24,>=20.9
-Requires-Dist: pandas <2,>=1.0.0
+Requires-Dist: pandas <3,>=1.0.0
 Requires-Dist: pyarrow
 Requires-Dist: pytimeparse <2,>=1.1.8
 Requires-Dist: pyyaml <7,>=6.0
@@ -255,7 +256,7 @@ Requires-Dist: sqlparse <1,>=0.4
 Requires-Dist: typing-extensions <5,>=4.1.0
 Requires-Dist: xgboost <2,>=1.7.3
 Provides-Extra: all
-Requires-Dist: lightgbm ==3.3.5 ; extra == 'all'
+Requires-Dist: lightgbm <4.2,>=3.3.5 ; 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'
@@ -263,11 +264,11 @@ Requires-Dist: sentencepiece <0.2,>=0.1.95 ; extra == 'all'
 Requires-Dist: shap ==0.42.1 ; extra == 'all'
 Requires-Dist: tensorflow <3,>=2.10 ; extra == 'all'
 Requires-Dist: tokenizers <1,>=0.10 ; extra == 'all'
+Requires-Dist: torch <3,>=2.0.1 ; extra == 'all'
 Requires-Dist: torchdata <1,>=0.4 ; extra == 'all'
 Requires-Dist: transformers <5,>=4.32.1 ; extra == 'all'
-Requires-Dist: torch <2.1.0,>=2.0.1 ; (platform_system == "Windows") and extra == 'all'
 Provides-Extra: lightgbm
-Requires-Dist: lightgbm ==3.3.5 ; extra == 'lightgbm'
+Requires-Dist: lightgbm <4.2,>=3.3.5 ; extra == 'lightgbm'
 Provides-Extra: llm
 Requires-Dist: peft <1,>=0.5.0 ; extra == 'llm'
 Provides-Extra: mlflow
@@ -277,8 +278,8 @@ Requires-Dist: shap ==0.42.1 ; extra == 'shap'
 Provides-Extra: tensorflow
 Requires-Dist: tensorflow <3,>=2.10 ; extra == 'tensorflow'
 Provides-Extra: torch
+Requires-Dist: torch <3,>=2.0.1 ; extra == 'torch'
 Requires-Dist: torchdata <1,>=0.4 ; extra == 'torch'
-Requires-Dist: torch <2.1.0,>=2.0.1 ; (platform_system == "Windows") and extra == 'torch'
 Provides-Extra: transformers
 Requires-Dist: sentence-transformers <3,>=2.2.2 ; extra == 'transformers'
 Requires-Dist: sentencepiece <0.2,>=0.1.95 ; extra == 'transformers'
@@ -370,13 +371,52 @@ be compatibility issues. Server-side functionality that `snowflake-ml-python` de
 
 # Release History
 
-## 1.3.1
+## 1.4.1
+
+### New Features
+
+- Registry: Add support for `catboost` model (`catboost.CatBoostClassifier`, `catboost.CatBoostRegressor`).
+- Registry: Add support for `lightgbm` model (`lightgbm.Booster`, `lightgbm.LightGBMClassifier`, `lightgbm.LightGBMRegressor`).
+
+### Bug Fixes
+
+- Registry: Fix a bug that leads to relax_version option is not working.
+
+## 1.4.0 (2024-04-08)
+
+### 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
 
@@ -412,8 +452,7 @@ be compatibility issues. Server-side functionality that `snowflake-ml-python` de
 - Registry: Improve the error message when specifying currently unsupported `pip_requirements` argument.
 - Model Development: Fix precision_recall_fscore_support incorrect results when `average="samples"`.
 - Model Registry: Fix an issue that leads to description, metrics or tags are not correctly returned in newly created
-  Model Registry (PrPr) due to Snowflake BCR [2024_01](
-  https://docs.snowflake.com/en/release-notes/bcr-bundles/2024_01/bcr-1483)
+  Model Registry (PrPr) due to Snowflake BCR [2024_01](https://docs.snowflake.com/en/release-notes/bcr-bundles/2024_01/bcr-1483)
 
 ### Behavior Changes
 
@@ -465,9 +504,9 @@ be compatibility issues. Server-side functionality that `snowflake-ml-python` de
 ### Additional Notes
 
 - Model Registry: The `snowflake.ml.registry.model_registry.ModelRegistry` has been deprecated starting from version
-1.2.0. It will stay in the Private Preview phase. For future implementations, kindly utilize
-`snowflake.ml.registry.Registry`, except when specifically required. The old model registry will be removed once all
-its primary functionalities are fully integrated into the new registry.
+  1.2.0. It will stay in the Private Preview phase. For future implementations, kindly utilize
+  `snowflake.ml.registry.Registry`, except when specifically required. The old model registry will be removed once all
+  its primary functionalities are fully integrated into the new registry.
 
 ## 1.1.2 (2023-12-18)
 
@@ -522,7 +561,7 @@ its primary functionalities are fully integrated into the new registry.
 - Model Registry: Fix regression issue that container logging is not shown during model deployment to SPCS.
 - Model Development: Enhance the column capacity of OrdinalEncoder.
 - Model Registry: Fix unbound `batch_size` error when deploying a model other than Hugging Face Pipeline
-   and LLM with GPU on SPCS.
+  and LLM with GPU on SPCS.
 
 ### Behavior Changes
 
@@ -564,7 +603,7 @@ its primary functionalities are fully integrated into the new registry.
 ### Behavior Changes
 
 - Model Development: precision_score, recall_score, f1_score, fbeta_score, precision_recall_fscore_support,
-mean_absolute_error, mean_squared_error, and mean_absolute_percentage_error metric calculations are now distributed.
+  mean_absolute_error, mean_squared_error, and mean_absolute_percentage_error metric calculations are now distributed.
 - Model Registry: `deploy` will now return `Deployment` for deployment information.
 
 ### New Features
@@ -578,7 +617,7 @@ mean_absolute_error, mean_squared_error, and mean_absolute_percentage_error metr
 - Model Development: Fix an issue that alias definitions cause `SnowparkSQLUnexpectedAliasException` in inference.
 - Model Registry: Fix an issue that signature inference could be incorrect when using Snowpark DataFrame as sample input.
 - Model Registry: Fix too strict data type validation when predicting. Now, for example, if you have a INT8
- type feature in the signature, if providing a INT64 dataframe but all values are within the range, it would not fail.
+  type feature in the signature, if providing a INT64 dataframe but all values are within the range, it would not fail.
 
 ## 1.0.9 (2023-09-28)
 
@@ -601,7 +640,7 @@ mean_absolute_error, mean_squared_error, and mean_absolute_percentage_error metr
 - Model Development: Fix an issue when the sklearn default value is `np.nan`.
 - Model Registry: Fix an issue that incorrect docker executable is used when building images.
 - Model Registry: Fix an issue that specifying `token` argument when using
-`snowflake.ml.model.models.huggingface_pipeline.HuggingFacePipelineModel` with `transformers < 4.32.0` is not effective.
+  `snowflake.ml.model.models.huggingface_pipeline.HuggingFacePipelineModel` with `transformers < 4.32.0` is not effective.
 - Model Registry: Fix an issue that incorrect system function call is used when deploying to SPCS.
 - Model Registry: Fix an issue when using a `transformers.pipeline` that does not have a `tokenizer`.
 - Model Registry: Fix incorrectly-inferred image repository name during model deployment to SPCS.
@@ -621,17 +660,17 @@ mean_absolute_error, mean_squared_error, and mean_absolute_percentage_error metr
 - Model Registry: add `create_if_not_exists` parameter in constructor.
 - Model Registry: Added get_or_create_model_registry API.
 - Model Registry: Added support for using GPU inference when deploying XGBoost (`xgboost.XGBModel` and `xgboost.Booster`
-), PyTorch (`torch.nn.Module` and `torch.jit.ScriptModule`) and TensorFlow (`tensorflow.Module` and
-`tensorflow.keras.Model`) models to Snowpark Container Services.
+  ), PyTorch (`torch.nn.Module` and `torch.jit.ScriptModule`) and TensorFlow (`tensorflow.Module` and
+  `tensorflow.keras.Model`) models to Snowpark Container Services.
 - Model Registry: When inferring model signature, `Sequence` of built-in types, `Sequence` of `numpy.ndarray`,
-`Sequence` of `torch.Tensor`, `Sequence` of `tensorflow.Tensor` and `Sequence` of `tensorflow.Tensor` can be used
- instead of only `List` of them.
+  `Sequence` of `torch.Tensor`, `Sequence` of `tensorflow.Tensor` and `Sequence` of `tensorflow.Tensor` can be used
+  instead of only `List` of them.
 - Model Registry: Added `get_training_dataset` API.
 - Model Development: Size of metrics result can exceed previous 8MB limit.
 - Model Registry: Added support save/load/deploy HuggingFace pipeline object (`transformers.Pipeline`) and our wrapper
-(`snowflake.ml.model.models.huggingface_pipeline.HuggingFacePipelineModel`) to it. Using the wrapper to specify
-configurations and the model for the pipeline will be loaded dynamically when deploying. Currently, following tasks
-are supported to log without manually specifying model signatures:
+  (`snowflake.ml.model.models.huggingface_pipeline.HuggingFacePipelineModel`) to it. Using the wrapper to specify
+  configurations and the model for the pipeline will be loaded dynamically when deploying. Currently, following tasks
+  are supported to log without manually specifying model signatures:
   - "conversational"
   - "fill-mask"
   - "question-answering"
@@ -655,18 +694,18 @@ are supported to log without manually specifying model signatures:
 - Model Registry: `log_model()` now return a `ModelReference` object instead of a model ID.
 - Model Registry: When deploying a model with 1 `target method` only, the `target_method` argument can be omitted.
 - Model Registry: When using the snowflake-ml-python with version newer than what is available in Snowflake Anaconda
-Channel, `embed_local_ml_library` option will be set as `True` automatically if not.
+  Channel, `embed_local_ml_library` option will be set as `True` automatically if not.
 - Model Registry: When deploying a model to Snowpark Container Services and using GPU, the default value of num_workers
-will be 1.
+  will be 1.
 - Model Registry: `keep_order` and `output_with_input_features` in the deploy options have been removed. Now the
-behavior is controlled by the type of the input when calling `model.predict()`. If the input is a `pandas.DataFrame`,
-the behavior will be the same as `keep_order=True` and `output_with_input_features=False` before. If the input is a
-`snowpark.DataFrame`, the behavior will be the same as `keep_order=False` and `output_with_input_features=True` before.
+  behavior is controlled by the type of the input when calling `model.predict()`. If the input is a `pandas.DataFrame`,
+  the behavior will be the same as `keep_order=True` and `output_with_input_features=False` before. If the input is a
+  `snowpark.DataFrame`, the behavior will be the same as `keep_order=False` and `output_with_input_features=True` before.
 - Model Registry: When logging and deploying PyTorch (`torch.nn.Module` and `torch.jit.ScriptModule`) and TensorFlow
-(`tensorflow.Module` and `tensorflow.keras.Model`) models, we no longer accept models whose input is a list of tensor
-and output is a list of tensors. Instead, now we accept models whose input is 1 or more tensors as positional arguments,
- and output is a tensor or a tuple of tensors. The input and output dataframe when predicting keep the same as before,
- that is every column is an array feature and contains a tensor.
+  (`tensorflow.Module` and `tensorflow.keras.Model`) models, we no longer accept models whose input is a list of tensor
+  and output is a list of tensors. Instead, now we accept models whose input is 1 or more tensors as positional arguments,
+  and output is a tensor or a tuple of tensors. The input and output dataframe when predicting keep the same as before,
+  that is every column is an array feature and contains a tensor.
 
 ## 1.0.5 (2023-08-17)
 
@@ -679,7 +718,7 @@ and output is a list of tensors. Instead, now we accept models whose input is 1
 
 - Model Registry: Restore the db/schema back to the session after `create_model_registry()`.
 - Model Registry: Fixed an issue that the UDF name created when deploying a model is not identical to what is provided
-and cannot be correctly dropped when deployment getting dropped.
+  and cannot be correctly dropped when deployment getting dropped.
 - connection_params.SnowflakeLoginOptions(): Added support for `private_key_path`.
 
 ## 1.0.4 (2023-07-28)
@@ -700,7 +739,7 @@ and cannot be correctly dropped when deployment getting dropped.
 ### Behavior Changes
 
 - Model Registry: When predicting a model whose output is a list of NumPy ndarray, the output would not be flattened,
-instead, every ndarray will act as a feature(column) in the output.
+  instead, every ndarray will act as a feature(column) in the output.
 
 ### New Features
 
@@ -709,18 +748,18 @@ instead, every ndarray will act as a feature(column) in the output.
 ### Bug Fixes
 
 - Model Registry: Fix an issue that when database or schema name provided to `create_model_registry` contains special
-characters, the model registry cannot be created.
+  characters, the model registry cannot be created.
 - Model Registry: Fix an issue that `get_model_description` returns with additional quotes.
 - Model Registry: Fix incorrect error message when attempting to remove a unset tag of a model.
 - Model Registry: Fix a typo in the default deployment table name.
 - Model Registry: Snowpark dataframe for sample input or input for `predict` method that contains a column with
-Snowflake `NUMBER(precision, scale)` data type where `scale = 0` will not lead to error, and will now correctly
-recognized as `INT64` data type in model signature.
+  Snowflake `NUMBER(precision, scale)` data type where `scale = 0` will not lead to error, and will now correctly
+  recognized as `INT64` data type in model signature.
 - Model Registry: Fix an issue that prevent model logged in the system whose default encoding is not UTF-8 compatible
-from deploying.
+  from deploying.
 - Model Registry: Added earlier and better error message when any file name in the model or the file name of model
-itself contains characters that are unable to be encoded using ASCII. It is currently not supported to deploy such a
-model.
+  itself contains characters that are unable to be encoded using ASCII. It is currently not supported to deploy such a
+  model.
 
 ## 1.0.2 (2023-06-22)
 
@@ -729,16 +768,16 @@ model.
 - Model Registry: Prohibit non-snowflake-native models from being logged.
 - Model Registry: `_use_local_snowml` parameter in options of `deploy()` has been removed.
 - Model Registry: A default `False` `embed_local_ml_library` parameter has been added to the options of `log_model()`.
-With this set to `False` (default), the version of the local snowflake-ml-python library will be recorded and used when
-deploying the model. With this set to `True`, local snowflake-ml-python library will be embedded into the logged model,
-and will be used when you load or deploy the model.
+  With this set to `False` (default), the version of the local snowflake-ml-python library will be recorded and used when
+  deploying the model. With this set to `True`, local snowflake-ml-python library will be embedded into the logged model,
+  and will be used when you load or deploy the model.
 
 ### New Features
 
 - Model Registry: A new optional argument named `code_paths` has been added to the arguments of `log_model()` for users
-to specify additional code paths to be imported when loading and deploying the model.
+  to specify additional code paths to be imported when loading and deploying the model.
 - Model Registry: A new optional argument named `options` has been added to the arguments of `log_model()` to specify
-any additional options when saving the model.
+  any additional options when saving the model.
 - Model Development: Added metrics:
   - d2_absolute_error_score
   - d2_pinball_score
@@ -757,14 +796,14 @@ any additional options when saving the model.
 
 - Model Development: Changed Metrics APIs to imitate sklearn metrics modules:
   - `accuracy_score()`, `confusion_matrix()`, `precision_recall_fscore_support()`, `precision_score()` methods move from
-  respective modules to `metrics.classification`.
+    respective modules to `metrics.classification`.
 - Model Registry: The default table/stage created by the Registry now uses "_SYSTEM_" as a prefix.
 - Model Registry: `get_model_history()` method as been enhanced to include the history of model deployment.
 
 ### New Features
 
 - Model Registry: A default `False` flag named `replace_udf` has been added to the options of `deploy()`. Setting this
-to `True` will allow overwrite existing UDF with the same name when deploying.
+  to `True` will allow overwrite existing UDF with the same name when deploying.
 - Model Development: Added metrics:
   - f1_score
   - fbeta_score
@@ -774,9 +813,9 @@ to `True` will allow overwrite existing UDF with the same name when deploying.
   - log_loss
   - precision_recall_curve
 - Model Registry: A new argument named `permanent` has been added to the argument of `deploy()`. Setting this to `True`
-allows the creation of a permanent deployment without needing to specify the UDF location.
+  allows the creation of a permanent deployment without needing to specify the UDF location.
 - Model Registry: A new method `list_deployments()` has been added to enumerate all permanent deployments originating
-from a specific model.
+  from a specific model.
 - Model Registry: A new method `get_deployment()` has been added to fetch a deployment by its deployment name.
 - Model Registry: A new method `delete_deployment()` has been added to remove an existing permanent deployment.
 
@@ -786,13 +825,13 @@ from a specific model.
 
 - Model Registry: `predict()` method moves from Registry to ModelReference.
 - Model Registry: `_snowml_wheel_path` parameter in options of `deploy()`, is replaced with `_use_local_snowml` with
-default value of `False`. Setting this to `True` will have the same effect of uploading local SnowML code when executing
-model in the warehouse.
+  default value of `False`. Setting this to `True` will have the same effect of uploading local SnowML code when executing
+  model in the warehouse.
 - Model Registry: Removed `id` field from `ModelReference` constructor.
 - Model Development: Preprocessing and Metrics move to the modeling package: `snowflake.ml.modeling.preprocessing` and
-`snowflake.ml.modeling.metrics`.
+  `snowflake.ml.modeling.metrics`.
 - Model Development: `get_sklearn_object()` method is renamed to `to_sklearn()`, `to_xgboost()`, and `to_lightgbm()` for
-respective native models.
+  respective native models.
 
 ### New Features
 
@@ -854,7 +893,7 @@ respective native models.
 ### New Features
 
 - Model Registry: Added support for delete_model. Use delete_artifact = False to not delete the underlying model data
-but just unregister.
+  but just unregister.
 
 ## 0.2.2 (2023-04-11)