snowflake-ml-python 1.11.0__py3-none-any.whl → 1.13.0__py3-none-any.whl

snowflake/ml/modeling/_internal/estimator_utils.py

@@ -42,6 +42,26 @@ def validate_sklearn_args(args: dict[str, tuple[Any, Any, bool]], klass: type) -
                     error_code=error_codes.DEPENDENCY_VERSION_ERROR,
                     original_exception=RuntimeError(f"Arg {k} is not supported by current version of SKLearn/XGBoost."),
                 )
+        elif v[0] == v[1] and v[0] != signature.parameters[k].default:
+            # If default value (pulled at autogen time) is not the same as the installed library's default value,
+            # we need to validate the parameter value against the parameter constraints.
+            # If the parameter value is invalid, we drop it.
+            try:
+                from sklearn.utils._param_validation import (
+                    InvalidParameterError,
+                    validate_parameter_constraints,
+                )
+
+                try:
+                    validate_parameter_constraints(
+                        klass._parameter_constraints,  # type: ignore[attr-defined]
+                        {k: v[0]},
+                        klass.__name__,
+                    )
+                except InvalidParameterError:
+                    continue  # Let the underlying estimator fill in the default value.
+            except (ImportError, AttributeError, TypeError):
+                result[k] = v[0]  # Try to use the value as is.
         else:
             result[k] = v[0]
     return result
@@ -199,7 +219,12 @@ def handle_inference_result(
         transformed_numpy_array = np.hstack(transformed_numpy_array)  # type: ignore[call-overload]
 
     if len(transformed_numpy_array.shape) == 1:
-        transformed_numpy_array = np.reshape(transformed_numpy_array, (-1, 1))
+        # Within a vectorized UDF, a single-row batch often yields a 1D array of length n_components.
+        # That must be reshaped to (1, n_components) to keep the number of rows aligned with the input batch.
+        if len(output_cols) > 1:
+            transformed_numpy_array = np.reshape(transformed_numpy_array, (1, -1))
+        else:
+            transformed_numpy_array = np.reshape(transformed_numpy_array, (-1, 1))
 
     shape = transformed_numpy_array.shape
     if len(shape) > 1:
@@ -292,3 +317,20 @@ def should_include_sample_weight(estimator: object, method_name: str) -> bool:
         return True
 
     return False
+
+
+def is_multi_task_estimator(estimator: object) -> bool:
+    """
+    Check if the estimator is a multi-task estimator that requires 2D targets.
+
+    Args:
+        estimator: The estimator to check
+
+    Returns:
+        True if the estimator is a multi-task estimator, False otherwise
+    """
+    # List of known multi-task estimators that require 2D targets
+    multi_task_estimators = {"MultiTaskElasticNet", "MultiTaskElasticNetCV", "MultiTaskLasso", "MultiTaskLassoCV"}
+
+    estimator_name = estimator.__class__.__name__
+    return estimator_name in multi_task_estimators

snowflake/ml/modeling/_internal/local_implementations/pandas_trainer.py

@@ -3,7 +3,10 @@ from typing import Optional
 
 import pandas as pd
 
-from snowflake.ml.modeling._internal.estimator_utils import handle_inference_result
+from snowflake.ml.modeling._internal.estimator_utils import (
+    handle_inference_result,
+    is_multi_task_estimator,
+)
 
 
 class PandasModelTrainer:
@@ -48,7 +51,11 @@ class PandasModelTrainer:
 
         if self.label_cols:
             label_arg_name = "Y" if "Y" in params else "y"
-            args[label_arg_name] = self.dataset[self.label_cols].squeeze()
+            # For multi-task estimators, avoid squeezing to maintain 2D shape
+            if is_multi_task_estimator(self.estimator):
+                args[label_arg_name] = self.dataset[self.label_cols]
+            else:
+                args[label_arg_name] = self.dataset[self.label_cols].squeeze()
 
         if self.sample_weight_col is not None and "sample_weight" in params:
             args["sample_weight"] = self.dataset[self.sample_weight_col].squeeze()
@@ -115,7 +122,11 @@ class PandasModelTrainer:
         args = {"X": self.dataset[self.input_cols]}
         if self.label_cols:
             label_arg_name = "Y" if "Y" in params else "y"
-            args[label_arg_name] = self.dataset[self.label_cols].squeeze()
+            # For multi-task estimators, avoid squeezing to maintain 2D shape
+            if is_multi_task_estimator(self.estimator):
+                args[label_arg_name] = self.dataset[self.label_cols]
+            else:
+                args[label_arg_name] = self.dataset[self.label_cols].squeeze()
 
         if self.sample_weight_col is not None and "sample_weight" in params:
             args["sample_weight"] = self.dataset[self.sample_weight_col].squeeze()

snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_trainer.py

@@ -22,6 +22,7 @@ from snowflake.ml._internal.utils import (
 from snowflake.ml.modeling._internal import estimator_utils
 from snowflake.ml.modeling._internal.estimator_utils import (
     handle_inference_result,
+    is_multi_task_estimator,
     should_include_sample_weight,
 )
 from snowflake.ml.modeling._internal.model_specifications import (
@@ -178,7 +179,11 @@ class SnowparkModelTrainer:
                 args = {"X": df[input_cols]}
                 if label_cols:
                     label_arg_name = "Y" if "Y" in params else "y"
-                    args[label_arg_name] = df[label_cols].squeeze()
+                    # For multi-task estimators, avoid squeezing to maintain 2D shape
+                    if is_multi_task_estimator(estimator):
+                        args[label_arg_name] = df[label_cols]
+                    else:
+                        args[label_arg_name] = df[label_cols].squeeze()
 
                 # Sample weight is not included in search estimators parameters, check the underlying estimator.
                 if sample_weight_col is not None and should_include_sample_weight(estimator, "fit"):
@@ -416,7 +421,11 @@ class SnowparkModelTrainer:
             args = {"X": df[input_cols]}
             if label_cols:
                 label_arg_name = "Y" if "Y" in params else "y"
-                args[label_arg_name] = df[label_cols].squeeze()
+                # For multi-task estimators, avoid squeezing to maintain 2D shape
+                if is_multi_task_estimator(estimator):
+                    args[label_arg_name] = df[label_cols]
+                else:
+                    args[label_arg_name] = df[label_cols].squeeze()
 
             if sample_weight_col is not None and should_include_sample_weight(estimator, "fit"):
                 args["sample_weight"] = df[sample_weight_col].squeeze()
@@ -734,12 +743,14 @@ class SnowparkModelTrainer:
         # Create a temp table in advance to store the output
         # This would allow us to use the same table outside the stored procedure
         df_one_line = dataset.limit(1).to_pandas(statement_params=statement_params)
-        df_one_line[
-            expected_output_cols_list[0]
-        ] = "[0]"  # Add one column as the output_col; this is a dummy value to represent the OBJECT type
+        # Pre-create ALL expected output columns so subsequent writes can target the same schema.
+        # Use a simple dummy string value to represent OBJECT-typed payloads.
+        for out_col in expected_output_cols_list:
+            df_one_line[out_col] = "[0]"
         if drop_input_cols:
+            # When input columns are dropped, the table should only contain the output columns.
             self.session.write_pandas(
-                df_one_line[expected_output_cols_list[0]],
+                df_one_line[expected_output_cols_list],
                 fit_transform_result_name,
                 auto_create_table=True,
                 table_type="temp",

snowflake/ml/modeling/calibration/calibrated_classifier_cv.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/affinity_propagation.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/agglomerative_clustering.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/birch.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/bisecting_k_means.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/dbscan.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/feature_agglomeration.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/k_means.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/mean_shift.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/mini_batch_k_means.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/optics.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/spectral_biclustering.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/spectral_clustering.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/cluster/spectral_coclustering.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/compose/column_transformer.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/compose/transformed_target_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/covariance/elliptic_envelope.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/covariance/empirical_covariance.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/covariance/graphical_lasso.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/covariance/graphical_lasso_cv.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/covariance/ledoit_wolf.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/covariance/min_cov_det.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/covariance/oas.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/covariance/shrunk_covariance.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/decomposition/dictionary_learning.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/decomposition/factor_analysis.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/decomposition/fast_ica.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/decomposition/incremental_pca.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/decomposition/kernel_pca.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/decomposition/mini_batch_dictionary_learning.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/decomposition/mini_batch_sparse_pca.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/decomposition/pca.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/decomposition/sparse_pca.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/decomposition/truncated_svd.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/discriminant_analysis/linear_discriminant_analysis.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/discriminant_analysis/quadratic_discriminant_analysis.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/ensemble/ada_boost_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/ensemble/ada_boost_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/ensemble/bagging_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/ensemble/bagging_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/ensemble/extra_trees_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/ensemble/extra_trees_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/ensemble/gradient_boosting_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/ensemble/gradient_boosting_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/ensemble/hist_gradient_boosting_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/ensemble/hist_gradient_boosting_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(