snowflake-ml-python 1.5.2__py3-none-any.whl → 1.5.3__py3-none-any.whl

snowflake/ml/modeling/preprocessing/robust_scaler.py

@@ -37,12 +37,15 @@ class RobustScaler(base.BaseTransformer):
             the dataset is scaled down. If less than 1, the dataset is scaled up.
 
         input_cols: Optional[Union[str, List[str]]], default=None
-            The name(s) of one or more columns in a DataFrame containing a feature to be scaled.
+            The name(s) of one or more columns in the input DataFrame containing feature(s) to be scaled. Input
+            columns must be specified before fit with this argument or after initialization with the
+            `set_input_cols` method. This argument is optional for API consistency.
 
         output_cols: Optional[Union[str, List[str]]], default=None
-            The name(s) of one or more columns in a DataFrame in which results will be stored. The number of
-            columns specified must match the number of input columns. For dense output, the column names specified are
-            used as base names for the columns created for each category.
+            The name(s) to assign output columns in the output DataFrame. The number of
+            columns specified must equal the number of input columns. Output columns must be specified before transform
+            with this argument or after initialization with the `set_output_cols` method. This argument is optional for
+            API consistency.
 
         passthrough_cols: Optional[Union[str, List[str]]], default=None
             A string or a list of strings indicating column names to be excluded from any

snowflake/ml/modeling/preprocessing/standard_scaler.py

@@ -26,11 +26,15 @@ class StandardScaler(base.BaseTransformer):
             If True, scale the data unit variance (i.e. unit standard deviation).
 
         input_cols: Optional[Union[str, List[str]]], default=None
-            The name(s) of one or more columns in a DataFrame containing a feature to be scaled.
+            The name(s) of one or more columns in the input DataFrame containing feature(s) to be scaled. Input
+            columns must be specified before fit with this argument or after initialization with the
+            `set_input_cols` method. This argument is optional for API consistency.
 
         output_cols: Optional[Union[str, List[str]]], default=None
-            The name(s) of one or more columns in a DataFrame in which results will be stored. The number of
-            columns specified must match the number of input columns.
+            The name(s) to assign output columns in the output DataFrame. The number of
+            columns specified must equal the number of input columns. Output columns must be specified before transform
+            with this argument or after initialization with the `set_output_cols` method. This argument is optional for
+            API consistency.
 
         passthrough_cols: Optional[Union[str, List[str]]], default=None
             A string or a list of strings indicating column names to be excluded from any

snowflake/ml/modeling/semi_supervised/label_propagation.py

@@ -257,7 +257,7 @@ class LabelPropagation(BaseTransformer):
                         inspect.currentframe(), LabelPropagation.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/semi_supervised/label_spreading.py

@@ -266,7 +266,7 @@ class LabelSpreading(BaseTransformer):
                         inspect.currentframe(), LabelSpreading.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/svm/linear_svc.py

@@ -322,7 +322,7 @@ class LinearSVC(BaseTransformer):
                         inspect.currentframe(), LinearSVC.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/svm/linear_svr.py

@@ -294,7 +294,7 @@ class LinearSVR(BaseTransformer):
                         inspect.currentframe(), LinearSVR.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/svm/nu_svc.py

@@ -328,7 +328,7 @@ class NuSVC(BaseTransformer):
                         inspect.currentframe(), NuSVC.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/svm/nu_svr.py

@@ -289,7 +289,7 @@ class NuSVR(BaseTransformer):
                         inspect.currentframe(), NuSVR.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/svm/svc.py

@@ -331,7 +331,7 @@ class SVC(BaseTransformer):
                         inspect.currentframe(), SVC.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/svm/svr.py

@@ -292,7 +292,7 @@ class SVR(BaseTransformer):
                         inspect.currentframe(), SVR.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/tree/decision_tree_classifier.py

@@ -359,7 +359,7 @@ class DecisionTreeClassifier(BaseTransformer):
                         inspect.currentframe(), DecisionTreeClassifier.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/tree/decision_tree_regressor.py

@@ -341,7 +341,7 @@ class DecisionTreeRegressor(BaseTransformer):
                         inspect.currentframe(), DecisionTreeRegressor.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/tree/extra_tree_classifier.py

@@ -351,7 +351,7 @@ class ExtraTreeClassifier(BaseTransformer):
                         inspect.currentframe(), ExtraTreeClassifier.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/tree/extra_tree_regressor.py

@@ -333,7 +333,7 @@ class ExtraTreeRegressor(BaseTransformer):
                         inspect.currentframe(), ExtraTreeRegressor.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/xgboost/xgb_classifier.py

@@ -451,7 +451,7 @@ class XGBClassifier(BaseTransformer):
                         inspect.currentframe(), XGBClassifier.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/xgboost/xgb_regressor.py

@@ -450,7 +450,7 @@ class XGBRegressor(BaseTransformer):
                         inspect.currentframe(), XGBRegressor.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/xgboost/xgbrf_classifier.py

@@ -455,7 +455,7 @@ class XGBRFClassifier(BaseTransformer):
                         inspect.currentframe(), XGBRFClassifier.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/modeling/xgboost/xgbrf_regressor.py

@@ -455,7 +455,7 @@ class XGBRFRegressor(BaseTransformer):
                         inspect.currentframe(), XGBRFRegressor.__class__.__name__
                     ),
                     api_calls=[Session.call],
-                    custom_tags=dict([("autogen", True)]) if self._autogenerated else None,
+                    custom_tags={"autogen": True} if self._autogenerated else None,
                 )
                 pd_df: pd.DataFrame = dataset.to_pandas(statement_params=statement_params)
                 pd_df.columns = dataset.columns

snowflake/ml/version.py

@@ -1 +1 @@
-VERSION="1.5.2"
+VERSION="1.5.3"

{snowflake_ml_python-1.5.2.dist-info → snowflake_ml_python-1.5.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: snowflake-ml-python
-Version: 1.5.2
+Version: 1.5.3
 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: 
@@ -250,7 +250,7 @@ Requires-Dist: s3fs <2024,>=2022.11
 Requires-Dist: scikit-learn <1.4,>=1.2.1
 Requires-Dist: scipy <2,>=1.9
 Requires-Dist: snowflake-connector-python[pandas] <4,>=3.5.0
-Requires-Dist: snowflake-snowpark-python !=1.12.0,<2,>=1.11.1
+Requires-Dist: snowflake-snowpark-python <2,>=1.15.0
 Requires-Dist: sqlparse <1,>=0.4
 Requires-Dist: typing-extensions <5,>=4.1.0
 Requires-Dist: xgboost <2,>=1.7.3
@@ -373,7 +373,23 @@ be compatibility issues. Server-side functionality that `snowflake-ml-python` de
 
 # Release History
 
-## 1.5.2
+## 1.5.3
+
+### Bug Fixes
+
+- Modeling: Fix an issue causing lineage information to be missing for
+  `Pipeline`, `GridSearchCV` , `SimpleImputer`, and `RandomizedSearchCV`
+- Registry: Fix an issue that leads to incorrect result when using pandas Dataframe with over 100, 000 rows as the input
+  of `ModelVersion.run` method in Stored Procedure.
+
+### Behavior Changes
+
+### New Features
+
+- Registry: Add support for TIMESTAMP_NTZ model signature data type, allowing timestamp input and output.
+- Dataset: Add `DatasetVersion.label_cols` and `DatasetVersion.exclude_cols` properties.
+
+## 1.5.2 (06-10-2024)
 
 ### Bug Fixes
 
@@ -385,7 +401,7 @@ be compatibility issues. Server-side functionality that `snowflake-ml-python` de
 
 ### New Features
 
-## 1.5.1
+## 1.5.1 (05-22-2024)
 
 ### Bug Fixes
 
@@ -410,7 +426,7 @@ be compatibility issues. Server-side functionality that `snowflake-ml-python` de
   permissions to operate on schema. Please call
   `import snowflake.ml.modeling.parameters.enable_anonymous_sproc  # noqa: F401`
 
-## 1.5.0
+## 1.5.0 (05-01-2024)
 
 ### Bug Fixes