snowflake-ml-python 1.6.2__py3-none-any.whl → 1.6.4__py3-none-any.whl

snowflake/ml/modeling/linear_model/ard_regression.py

@@ -567,6 +567,7 @@ class ARDRegression(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/bayesian_ridge.py

@@ -578,6 +578,7 @@ class BayesianRidge(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/elastic_net.py

@@ -577,6 +577,7 @@ class ElasticNet(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/elastic_net_cv.py

@@ -613,6 +613,7 @@ class ElasticNetCV(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/gamma_regressor.py

@@ -558,6 +558,7 @@ class GammaRegressor(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/huber_regressor.py

@@ -541,6 +541,7 @@ class HuberRegressor(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/lars.py

@@ -570,6 +570,7 @@ class Lars(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/lars_cv.py

@@ -578,6 +578,7 @@ class LarsCV(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/lasso.py

@@ -571,6 +571,7 @@ class Lasso(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/lasso_cv.py

@@ -599,6 +599,7 @@ class LassoCV(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/lasso_lars.py

@@ -591,6 +591,7 @@ class LassoLars(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/lasso_lars_cv.py

@@ -592,6 +592,7 @@ class LassoLarsCV(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/lasso_lars_ic.py

@@ -575,6 +575,7 @@ class LassoLarsIC(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/linear_regression.py

@@ -528,6 +528,7 @@ class LinearRegression(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/logistic_regression.py

@@ -642,6 +642,7 @@ class LogisticRegression(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/logistic_regression_cv.py

@@ -663,6 +663,7 @@ class LogisticRegressionCV(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/multi_task_elastic_net.py

@@ -561,6 +561,7 @@ class MultiTaskElasticNet(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/multi_task_elastic_net_cv.py

@@ -602,6 +602,7 @@ class MultiTaskElasticNetCV(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/multi_task_lasso.py

@@ -553,6 +553,7 @@ class MultiTaskLasso(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/multi_task_lasso_cv.py

@@ -588,6 +588,7 @@ class MultiTaskLassoCV(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/orthogonal_matching_pursuit.py

@@ -536,6 +536,7 @@ class OrthogonalMatchingPursuit(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/passive_aggressive_classifier.py

@@ -610,6 +610,7 @@ class PassiveAggressiveClassifier(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/passive_aggressive_regressor.py

@@ -596,6 +596,7 @@ class PassiveAggressiveRegressor(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/perceptron.py

@@ -609,6 +609,7 @@ class Perceptron(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/poisson_regressor.py

@@ -558,6 +558,7 @@ class PoissonRegressor(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/ransac_regressor.py

@@ -614,6 +614,7 @@ class RANSACRegressor(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/ridge.py

@@ -606,6 +606,7 @@ class Ridge(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/ridge_classifier.py

@@ -606,6 +606,7 @@ class RidgeClassifier(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/ridge_classifier_cv.py

@@ -557,6 +557,7 @@ class RidgeClassifierCV(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/ridge_cv.py

@@ -578,6 +578,7 @@ class RidgeCV(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/sgd_classifier.py

@@ -697,6 +697,7 @@ class SGDClassifier(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/sgd_one_class_svm.py

@@ -599,6 +599,7 @@ class SGDOneClassSVM(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/sgd_regressor.py

@@ -663,6 +663,7 @@ class SGDRegressor(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/theil_sen_regressor.py

@@ -565,6 +565,7 @@ class TheilSenRegressor(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/linear_model/tweedie_regressor.py

@@ -591,6 +591,7 @@ class TweedieRegressor(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/manifold/isomap.py

@@ -591,6 +591,7 @@ class Isomap(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/manifold/mds.py

@@ -572,6 +572,7 @@ class MDS(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/manifold/spectral_embedding.py

@@ -574,6 +574,7 @@ class SpectralEmbedding(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/manifold/tsne.py

@@ -633,6 +633,7 @@ class TSNE(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/metrics/metrics_utils.py

@@ -59,7 +59,7 @@ def register_accumulator_udtf(*, session: Session, statement_params: Dict[str, A
             ]
         ),
         input_types=[T.BinaryType()],
-        packages=["numpy", "cloudpickle"],
+        packages=[f"numpy=={np.__version__}", f"cloudpickle=={cloudpickle.__version__}"],
         name=accumulator,
         is_permanent=False,
         replace=True,
@@ -174,7 +174,7 @@ def register_sharded_dot_sum_computer(*, session: Session, statement_params: Dic
             ]
         ),
         input_types=[T.ArrayType(), T.IntegerType(), T.IntegerType()],
-        packages=["numpy", "cloudpickle"],
+        packages=[f"numpy=={np.__version__}", f"cloudpickle=={cloudpickle.__version__}"],
         name=sharded_dot_and_sum_computer,
         is_permanent=False,
         replace=True,

snowflake/ml/modeling/metrics/ranking.py

@@ -102,7 +102,6 @@ def precision_recall_curve(
         ],
         statement_params=statement_params,
         anonymous=True,
-        execute_as="caller",
     )
     def precision_recall_curve_anon_sproc(session: snowpark.Session) -> bytes:
         for query in queries[:-1]:
@@ -250,7 +249,6 @@ def roc_auc_score(
         ],
         statement_params=statement_params,
         anonymous=True,
-        execute_as="caller",
     )
     def roc_auc_score_anon_sproc(session: snowpark.Session) -> bytes:
         for query in queries[:-1]:
@@ -354,7 +352,6 @@ def roc_curve(
         ],
         statement_params=statement_params,
         anonymous=True,
-        execute_as="caller",
     )
     def roc_curve_anon_sproc(session: snowpark.Session) -> bytes:
         for query in queries[:-1]:

snowflake/ml/modeling/metrics/regression.py

@@ -87,7 +87,6 @@ def d2_absolute_error_score(
         ],
         statement_params=statement_params,
         anonymous=True,
-        execute_as="caller",
     )
     def d2_absolute_error_score_anon_sproc(session: snowpark.Session) -> bytes:
         for query in queries[:-1]:
@@ -185,7 +184,6 @@ def d2_pinball_score(
         ],
         statement_params=statement_params,
         anonymous=True,
-        execute_as="caller",
     )
     def d2_pinball_score_anon_sproc(session: snowpark.Session) -> bytes:
         for query in queries[:-1]:
@@ -301,7 +299,6 @@ def explained_variance_score(
         ],
         statement_params=statement_params,
         anonymous=True,
-        execute_as="caller",
     )
     def explained_variance_score_anon_sproc(session: snowpark.Session) -> bytes:
         for query in queries[:-1]:

snowflake/ml/modeling/mixture/bayesian_gaussian_mixture.py

@@ -638,6 +638,7 @@ class BayesianGaussianMixture(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/mixture/gaussian_mixture.py

@@ -611,6 +611,7 @@ class GaussianMixture(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/multiclass/one_vs_one_classifier.py

@@ -519,6 +519,7 @@ class OneVsOneClassifier(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/multiclass/one_vs_rest_classifier.py

@@ -528,6 +528,7 @@ class OneVsRestClassifier(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/multiclass/output_code_classifier.py

@@ -531,6 +531,7 @@ class OutputCodeClassifier(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/naive_bayes/bernoulli_nb.py

@@ -531,6 +531,7 @@ class BernoulliNB(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/naive_bayes/categorical_nb.py

@@ -537,6 +537,7 @@ class CategoricalNB(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/naive_bayes/complement_nb.py

@@ -531,6 +531,7 @@ class ComplementNB(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/naive_bayes/gaussian_nb.py

@@ -512,6 +512,7 @@ class GaussianNB(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/naive_bayes/multinomial_nb.py

@@ -525,6 +525,7 @@ class MultinomialNB(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/neighbors/k_neighbors_classifier.py

@@ -582,6 +582,7 @@ class KNeighborsClassifier(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/neighbors/k_neighbors_regressor.py

@@ -584,6 +584,7 @@ class KNeighborsRegressor(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/neighbors/kernel_density.py

@@ -561,6 +561,7 @@ class KernelDensity(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/neighbors/local_outlier_factor.py

@@ -593,6 +593,7 @@ class LocalOutlierFactor(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/neighbors/nearest_centroid.py

@@ -522,6 +522,7 @@ class NearestCentroid(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/neighbors/nearest_neighbors.py

@@ -572,6 +572,7 @@ class NearestNeighbors(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/neighbors/neighborhood_components_analysis.py

@@ -595,6 +595,7 @@ class NeighborhoodComponentsAnalysis(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/neighbors/radius_neighbors_classifier.py

@@ -594,6 +594,7 @@ class RadiusNeighborsClassifier(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/neighbors/radius_neighbors_regressor.py

@@ -584,6 +584,7 @@ class RadiusNeighborsRegressor(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/neural_network/bernoulli_rbm.py

@@ -545,6 +545,7 @@ class BernoulliRBM(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/neural_network/mlp_classifier.py

@@ -696,6 +696,7 @@ class MLPClassifier(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,

snowflake/ml/modeling/neural_network/mlp_regressor.py

@@ -692,6 +692,7 @@ class MLPRegressor(BaseTransformer):
         """
         self._infer_input_output_cols(dataset)
         super()._check_dataset_type(dataset)
+
         model_trainer = ModelTrainerBuilder.build_fit_transform(
             estimator=self._sklearn_object,
             dataset=dataset,