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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (262) hide show
  1. snowflake/cortex/__init__.py +4 -0
  2. snowflake/cortex/_classify_text.py +2 -2
  3. snowflake/cortex/_embed_text_1024.py +37 -0
  4. snowflake/cortex/_embed_text_768.py +37 -0
  5. snowflake/cortex/_extract_answer.py +2 -2
  6. snowflake/cortex/_sentiment.py +2 -2
  7. snowflake/cortex/_summarize.py +2 -2
  8. snowflake/cortex/_translate.py +2 -2
  9. snowflake/cortex/_util.py +4 -4
  10. snowflake/ml/_internal/env_utils.py +5 -5
  11. snowflake/ml/_internal/exceptions/error_codes.py +2 -0
  12. snowflake/ml/_internal/utils/db_utils.py +50 -0
  13. snowflake/ml/_internal/utils/service_logger.py +63 -0
  14. snowflake/ml/_internal/utils/sql_identifier.py +25 -1
  15. snowflake/ml/data/_internal/arrow_ingestor.py +1 -11
  16. snowflake/ml/data/ingestor_utils.py +20 -10
  17. snowflake/ml/feature_store/access_manager.py +3 -3
  18. snowflake/ml/feature_store/feature_store.py +19 -2
  19. snowflake/ml/feature_store/feature_view.py +82 -28
  20. snowflake/ml/fileset/stage_fs.py +2 -1
  21. snowflake/ml/lineage/lineage_node.py +7 -2
  22. snowflake/ml/model/__init__.py +1 -2
  23. snowflake/ml/model/_client/model/model_version_impl.py +78 -9
  24. snowflake/ml/model/_client/ops/model_ops.py +89 -7
  25. snowflake/ml/model/_client/ops/service_ops.py +200 -91
  26. snowflake/ml/model/_client/service/model_deployment_spec.py +4 -0
  27. snowflake/ml/model/_client/service/model_deployment_spec_schema.py +1 -0
  28. snowflake/ml/model/_client/sql/_base.py +5 -0
  29. snowflake/ml/model/_client/sql/model.py +1 -0
  30. snowflake/ml/model/_client/sql/model_version.py +9 -5
  31. snowflake/ml/model/_client/sql/service.py +35 -13
  32. snowflake/ml/model/_model_composer/model_composer.py +11 -41
  33. snowflake/ml/model/_model_composer/model_manifest/model_manifest.py +29 -4
  34. snowflake/ml/model/_packager/model_env/model_env.py +4 -38
  35. snowflake/ml/model/_packager/model_handlers/_utils.py +106 -32
  36. snowflake/ml/model/_packager/model_handlers/catboost.py +26 -27
  37. snowflake/ml/model/_packager/model_handlers/huggingface_pipeline.py +3 -3
  38. snowflake/ml/model/_packager/model_handlers/lightgbm.py +21 -6
  39. snowflake/ml/model/_packager/model_handlers/mlflow.py +3 -5
  40. snowflake/ml/model/_packager/model_handlers/model_objective_utils.py +111 -58
  41. snowflake/ml/model/_packager/model_handlers/sentence_transformers.py +15 -8
  42. snowflake/ml/model/_packager/model_handlers/sklearn.py +50 -66
  43. snowflake/ml/model/_packager/model_handlers/snowmlmodel.py +36 -17
  44. snowflake/ml/model/_packager/model_handlers/xgboost.py +22 -7
  45. snowflake/ml/model/_packager/model_meta/model_meta.py +16 -45
  46. snowflake/ml/model/_packager/model_meta/model_meta_schema.py +1 -6
  47. snowflake/ml/model/_packager/model_packager.py +14 -10
  48. snowflake/ml/model/_packager/model_runtime/model_runtime.py +11 -0
  49. snowflake/ml/model/_signatures/snowpark_handler.py +3 -2
  50. snowflake/ml/model/type_hints.py +11 -152
  51. snowflake/ml/modeling/_internal/snowpark_implementations/distributed_hpo_trainer.py +0 -2
  52. snowflake/ml/modeling/_internal/snowpark_implementations/snowpark_trainer.py +17 -6
  53. snowflake/ml/modeling/_internal/snowpark_implementations/xgboost_external_memory_trainer.py +0 -1
  54. snowflake/ml/modeling/calibration/calibrated_classifier_cv.py +1 -0
  55. snowflake/ml/modeling/cluster/affinity_propagation.py +1 -0
  56. snowflake/ml/modeling/cluster/agglomerative_clustering.py +1 -0
  57. snowflake/ml/modeling/cluster/birch.py +1 -0
  58. snowflake/ml/modeling/cluster/bisecting_k_means.py +1 -0
  59. snowflake/ml/modeling/cluster/dbscan.py +1 -0
  60. snowflake/ml/modeling/cluster/feature_agglomeration.py +1 -0
  61. snowflake/ml/modeling/cluster/k_means.py +1 -0
  62. snowflake/ml/modeling/cluster/mean_shift.py +1 -0
  63. snowflake/ml/modeling/cluster/mini_batch_k_means.py +1 -0
  64. snowflake/ml/modeling/cluster/optics.py +1 -0
  65. snowflake/ml/modeling/cluster/spectral_biclustering.py +1 -0
  66. snowflake/ml/modeling/cluster/spectral_clustering.py +1 -0
  67. snowflake/ml/modeling/cluster/spectral_coclustering.py +1 -0
  68. snowflake/ml/modeling/compose/column_transformer.py +1 -0
  69. snowflake/ml/modeling/compose/transformed_target_regressor.py +1 -0
  70. snowflake/ml/modeling/covariance/elliptic_envelope.py +1 -0
  71. snowflake/ml/modeling/covariance/empirical_covariance.py +1 -0
  72. snowflake/ml/modeling/covariance/graphical_lasso.py +1 -0
  73. snowflake/ml/modeling/covariance/graphical_lasso_cv.py +1 -0
  74. snowflake/ml/modeling/covariance/ledoit_wolf.py +1 -0
  75. snowflake/ml/modeling/covariance/min_cov_det.py +1 -0
  76. snowflake/ml/modeling/covariance/oas.py +1 -0
  77. snowflake/ml/modeling/covariance/shrunk_covariance.py +1 -0
  78. snowflake/ml/modeling/decomposition/dictionary_learning.py +1 -0
  79. snowflake/ml/modeling/decomposition/factor_analysis.py +1 -0
  80. snowflake/ml/modeling/decomposition/fast_ica.py +1 -0
  81. snowflake/ml/modeling/decomposition/incremental_pca.py +1 -0
  82. snowflake/ml/modeling/decomposition/kernel_pca.py +1 -0
  83. snowflake/ml/modeling/decomposition/mini_batch_dictionary_learning.py +1 -0
  84. snowflake/ml/modeling/decomposition/mini_batch_sparse_pca.py +1 -0
  85. snowflake/ml/modeling/decomposition/pca.py +1 -0
  86. snowflake/ml/modeling/decomposition/sparse_pca.py +1 -0
  87. snowflake/ml/modeling/decomposition/truncated_svd.py +1 -0
  88. snowflake/ml/modeling/discriminant_analysis/linear_discriminant_analysis.py +1 -0
  89. snowflake/ml/modeling/discriminant_analysis/quadratic_discriminant_analysis.py +1 -0
  90. snowflake/ml/modeling/ensemble/ada_boost_classifier.py +1 -0
  91. snowflake/ml/modeling/ensemble/ada_boost_regressor.py +1 -0
  92. snowflake/ml/modeling/ensemble/bagging_classifier.py +1 -0
  93. snowflake/ml/modeling/ensemble/bagging_regressor.py +1 -0
  94. snowflake/ml/modeling/ensemble/extra_trees_classifier.py +1 -0
  95. snowflake/ml/modeling/ensemble/extra_trees_regressor.py +1 -0
  96. snowflake/ml/modeling/ensemble/gradient_boosting_classifier.py +1 -0
  97. snowflake/ml/modeling/ensemble/gradient_boosting_regressor.py +1 -0
  98. snowflake/ml/modeling/ensemble/hist_gradient_boosting_classifier.py +1 -0
  99. snowflake/ml/modeling/ensemble/hist_gradient_boosting_regressor.py +1 -0
  100. snowflake/ml/modeling/ensemble/isolation_forest.py +1 -0
  101. snowflake/ml/modeling/ensemble/random_forest_classifier.py +1 -0
  102. snowflake/ml/modeling/ensemble/random_forest_regressor.py +1 -0
  103. snowflake/ml/modeling/ensemble/stacking_regressor.py +1 -0
  104. snowflake/ml/modeling/ensemble/voting_classifier.py +1 -0
  105. snowflake/ml/modeling/ensemble/voting_regressor.py +1 -0
  106. snowflake/ml/modeling/feature_selection/generic_univariate_select.py +1 -0
  107. snowflake/ml/modeling/feature_selection/select_fdr.py +1 -0
  108. snowflake/ml/modeling/feature_selection/select_fpr.py +1 -0
  109. snowflake/ml/modeling/feature_selection/select_fwe.py +1 -0
  110. snowflake/ml/modeling/feature_selection/select_k_best.py +1 -0
  111. snowflake/ml/modeling/feature_selection/select_percentile.py +1 -0
  112. snowflake/ml/modeling/feature_selection/sequential_feature_selector.py +1 -0
  113. snowflake/ml/modeling/feature_selection/variance_threshold.py +1 -0
  114. snowflake/ml/modeling/gaussian_process/gaussian_process_classifier.py +1 -0
  115. snowflake/ml/modeling/gaussian_process/gaussian_process_regressor.py +1 -0
  116. snowflake/ml/modeling/impute/iterative_imputer.py +1 -0
  117. snowflake/ml/modeling/impute/knn_imputer.py +1 -0
  118. snowflake/ml/modeling/impute/missing_indicator.py +1 -0
  119. snowflake/ml/modeling/kernel_approximation/additive_chi2_sampler.py +1 -0
  120. snowflake/ml/modeling/kernel_approximation/nystroem.py +1 -0
  121. snowflake/ml/modeling/kernel_approximation/polynomial_count_sketch.py +1 -0
  122. snowflake/ml/modeling/kernel_approximation/rbf_sampler.py +1 -0
  123. snowflake/ml/modeling/kernel_approximation/skewed_chi2_sampler.py +1 -0
  124. snowflake/ml/modeling/kernel_ridge/kernel_ridge.py +1 -0
  125. snowflake/ml/modeling/lightgbm/lgbm_classifier.py +1 -0
  126. snowflake/ml/modeling/lightgbm/lgbm_regressor.py +1 -0
  127. snowflake/ml/modeling/linear_model/ard_regression.py +1 -0
  128. snowflake/ml/modeling/linear_model/bayesian_ridge.py +1 -0
  129. snowflake/ml/modeling/linear_model/elastic_net.py +1 -0
  130. snowflake/ml/modeling/linear_model/elastic_net_cv.py +1 -0
  131. snowflake/ml/modeling/linear_model/gamma_regressor.py +1 -0
  132. snowflake/ml/modeling/linear_model/huber_regressor.py +1 -0
  133. snowflake/ml/modeling/linear_model/lars.py +1 -0
  134. snowflake/ml/modeling/linear_model/lars_cv.py +1 -0
  135. snowflake/ml/modeling/linear_model/lasso.py +1 -0
  136. snowflake/ml/modeling/linear_model/lasso_cv.py +1 -0
  137. snowflake/ml/modeling/linear_model/lasso_lars.py +1 -0
  138. snowflake/ml/modeling/linear_model/lasso_lars_cv.py +1 -0
  139. snowflake/ml/modeling/linear_model/lasso_lars_ic.py +1 -0
  140. snowflake/ml/modeling/linear_model/linear_regression.py +1 -0
  141. snowflake/ml/modeling/linear_model/logistic_regression.py +1 -0
  142. snowflake/ml/modeling/linear_model/logistic_regression_cv.py +1 -0
  143. snowflake/ml/modeling/linear_model/multi_task_elastic_net.py +1 -0
  144. snowflake/ml/modeling/linear_model/multi_task_elastic_net_cv.py +1 -0
  145. snowflake/ml/modeling/linear_model/multi_task_lasso.py +1 -0
  146. snowflake/ml/modeling/linear_model/multi_task_lasso_cv.py +1 -0
  147. snowflake/ml/modeling/linear_model/orthogonal_matching_pursuit.py +1 -0
  148. snowflake/ml/modeling/linear_model/passive_aggressive_classifier.py +1 -0
  149. snowflake/ml/modeling/linear_model/passive_aggressive_regressor.py +1 -0
  150. snowflake/ml/modeling/linear_model/perceptron.py +1 -0
  151. snowflake/ml/modeling/linear_model/poisson_regressor.py +1 -0
  152. snowflake/ml/modeling/linear_model/ransac_regressor.py +1 -0
  153. snowflake/ml/modeling/linear_model/ridge.py +1 -0
  154. snowflake/ml/modeling/linear_model/ridge_classifier.py +1 -0
  155. snowflake/ml/modeling/linear_model/ridge_classifier_cv.py +1 -0
  156. snowflake/ml/modeling/linear_model/ridge_cv.py +1 -0
  157. snowflake/ml/modeling/linear_model/sgd_classifier.py +1 -0
  158. snowflake/ml/modeling/linear_model/sgd_one_class_svm.py +1 -0
  159. snowflake/ml/modeling/linear_model/sgd_regressor.py +1 -0
  160. snowflake/ml/modeling/linear_model/theil_sen_regressor.py +1 -0
  161. snowflake/ml/modeling/linear_model/tweedie_regressor.py +1 -0
  162. snowflake/ml/modeling/manifold/isomap.py +1 -0
  163. snowflake/ml/modeling/manifold/mds.py +1 -0
  164. snowflake/ml/modeling/manifold/spectral_embedding.py +1 -0
  165. snowflake/ml/modeling/manifold/tsne.py +1 -0
  166. snowflake/ml/modeling/metrics/metrics_utils.py +2 -2
  167. snowflake/ml/modeling/metrics/ranking.py +0 -3
  168. snowflake/ml/modeling/metrics/regression.py +0 -3
  169. snowflake/ml/modeling/mixture/bayesian_gaussian_mixture.py +1 -0
  170. snowflake/ml/modeling/mixture/gaussian_mixture.py +1 -0
  171. snowflake/ml/modeling/multiclass/one_vs_one_classifier.py +1 -0
  172. snowflake/ml/modeling/multiclass/one_vs_rest_classifier.py +1 -0
  173. snowflake/ml/modeling/multiclass/output_code_classifier.py +1 -0
  174. snowflake/ml/modeling/naive_bayes/bernoulli_nb.py +1 -0
  175. snowflake/ml/modeling/naive_bayes/categorical_nb.py +1 -0
  176. snowflake/ml/modeling/naive_bayes/complement_nb.py +1 -0
  177. snowflake/ml/modeling/naive_bayes/gaussian_nb.py +1 -0
  178. snowflake/ml/modeling/naive_bayes/multinomial_nb.py +1 -0
  179. snowflake/ml/modeling/neighbors/k_neighbors_classifier.py +1 -0
  180. snowflake/ml/modeling/neighbors/k_neighbors_regressor.py +1 -0
  181. snowflake/ml/modeling/neighbors/kernel_density.py +1 -0
  182. snowflake/ml/modeling/neighbors/local_outlier_factor.py +1 -0
  183. snowflake/ml/modeling/neighbors/nearest_centroid.py +1 -0
  184. snowflake/ml/modeling/neighbors/nearest_neighbors.py +1 -0
  185. snowflake/ml/modeling/neighbors/neighborhood_components_analysis.py +1 -0
  186. snowflake/ml/modeling/neighbors/radius_neighbors_classifier.py +1 -0
  187. snowflake/ml/modeling/neighbors/radius_neighbors_regressor.py +1 -0
  188. snowflake/ml/modeling/neural_network/bernoulli_rbm.py +1 -0
  189. snowflake/ml/modeling/neural_network/mlp_classifier.py +1 -0
  190. snowflake/ml/modeling/neural_network/mlp_regressor.py +1 -0
  191. snowflake/ml/modeling/pipeline/pipeline.py +0 -1
  192. snowflake/ml/modeling/preprocessing/polynomial_features.py +1 -0
  193. snowflake/ml/modeling/semi_supervised/label_propagation.py +1 -0
  194. snowflake/ml/modeling/semi_supervised/label_spreading.py +1 -0
  195. snowflake/ml/modeling/svm/linear_svc.py +1 -0
  196. snowflake/ml/modeling/svm/linear_svr.py +1 -0
  197. snowflake/ml/modeling/svm/nu_svc.py +1 -0
  198. snowflake/ml/modeling/svm/nu_svr.py +1 -0
  199. snowflake/ml/modeling/svm/svc.py +1 -0
  200. snowflake/ml/modeling/svm/svr.py +1 -0
  201. snowflake/ml/modeling/tree/decision_tree_classifier.py +1 -0
  202. snowflake/ml/modeling/tree/decision_tree_regressor.py +1 -0
  203. snowflake/ml/modeling/tree/extra_tree_classifier.py +1 -0
  204. snowflake/ml/modeling/tree/extra_tree_regressor.py +1 -0
  205. snowflake/ml/modeling/xgboost/xgb_classifier.py +1 -0
  206. snowflake/ml/modeling/xgboost/xgb_regressor.py +1 -0
  207. snowflake/ml/modeling/xgboost/xgbrf_classifier.py +1 -0
  208. snowflake/ml/modeling/xgboost/xgbrf_regressor.py +1 -0
  209. snowflake/ml/monitoring/_client/model_monitor.py +126 -0
  210. snowflake/ml/monitoring/_client/model_monitor_manager.py +361 -0
  211. snowflake/ml/monitoring/_client/model_monitor_version.py +1 -0
  212. snowflake/ml/monitoring/_client/monitor_sql_client.py +1335 -0
  213. snowflake/ml/monitoring/_client/queries/record_count.ssql +14 -0
  214. snowflake/ml/monitoring/_client/queries/rmse.ssql +28 -0
  215. snowflake/ml/monitoring/entities/model_monitor_config.py +28 -0
  216. snowflake/ml/monitoring/entities/model_monitor_interval.py +46 -0
  217. snowflake/ml/monitoring/entities/output_score_type.py +90 -0
  218. snowflake/ml/registry/_manager/model_manager.py +4 -4
  219. snowflake/ml/registry/registry.py +165 -6
  220. snowflake/ml/version.py +1 -1
  221. {snowflake_ml_python-1.6.2.dist-info → snowflake_ml_python-1.6.4.dist-info}/METADATA +30 -9
  222. {snowflake_ml_python-1.6.2.dist-info → snowflake_ml_python-1.6.4.dist-info}/RECORD +225 -249
  223. {snowflake_ml_python-1.6.2.dist-info → snowflake_ml_python-1.6.4.dist-info}/WHEEL +1 -1
  224. snowflake/ml/_internal/container_services/image_registry/credential.py +0 -84
  225. snowflake/ml/_internal/container_services/image_registry/http_client.py +0 -127
  226. snowflake/ml/_internal/container_services/image_registry/imagelib.py +0 -400
  227. snowflake/ml/_internal/container_services/image_registry/registry_client.py +0 -212
  228. snowflake/ml/_internal/utils/log_stream_processor.py +0 -30
  229. snowflake/ml/_internal/utils/session_token_manager.py +0 -46
  230. snowflake/ml/_internal/utils/spcs_attribution_utils.py +0 -122
  231. snowflake/ml/_internal/utils/uri.py +0 -77
  232. snowflake/ml/model/_api.py +0 -568
  233. snowflake/ml/model/_deploy_client/image_builds/base_image_builder.py +0 -12
  234. snowflake/ml/model/_deploy_client/image_builds/client_image_builder.py +0 -249
  235. snowflake/ml/model/_deploy_client/image_builds/docker_context.py +0 -130
  236. snowflake/ml/model/_deploy_client/image_builds/gunicorn_run.sh +0 -36
  237. snowflake/ml/model/_deploy_client/image_builds/inference_server/main.py +0 -268
  238. snowflake/ml/model/_deploy_client/image_builds/server_image_builder.py +0 -215
  239. snowflake/ml/model/_deploy_client/image_builds/templates/dockerfile_template +0 -53
  240. snowflake/ml/model/_deploy_client/image_builds/templates/image_build_job_spec_template +0 -38
  241. snowflake/ml/model/_deploy_client/image_builds/templates/kaniko_shell_script_template +0 -105
  242. snowflake/ml/model/_deploy_client/snowservice/deploy.py +0 -611
  243. snowflake/ml/model/_deploy_client/snowservice/deploy_options.py +0 -116
  244. snowflake/ml/model/_deploy_client/snowservice/instance_types.py +0 -10
  245. snowflake/ml/model/_deploy_client/snowservice/templates/service_spec_template +0 -28
  246. snowflake/ml/model/_deploy_client/snowservice/templates/service_spec_template_with_model +0 -21
  247. snowflake/ml/model/_deploy_client/utils/constants.py +0 -48
  248. snowflake/ml/model/_deploy_client/utils/snowservice_client.py +0 -280
  249. snowflake/ml/model/_deploy_client/warehouse/deploy.py +0 -202
  250. snowflake/ml/model/_deploy_client/warehouse/infer_template.py +0 -99
  251. snowflake/ml/model/_packager/model_handlers/llm.py +0 -269
  252. snowflake/ml/model/_packager/model_meta/_core_requirements.py +0 -11
  253. snowflake/ml/model/deploy_platforms.py +0 -6
  254. snowflake/ml/model/models/llm.py +0 -106
  255. snowflake/ml/monitoring/monitor.py +0 -203
  256. snowflake/ml/registry/_initial_schema.py +0 -142
  257. snowflake/ml/registry/_schema.py +0 -82
  258. snowflake/ml/registry/_schema_upgrade_plans.py +0 -116
  259. snowflake/ml/registry/_schema_version_manager.py +0 -163
  260. snowflake/ml/registry/model_registry.py +0 -2048
  261. {snowflake_ml_python-1.6.2.dist-info → snowflake_ml_python-1.6.4.dist-info}/LICENSE.txt +0 -0
  262. {snowflake_ml_python-1.6.2.dist-info → snowflake_ml_python-1.6.4.dist-info}/top_level.txt +0 -0
snowflake/ml/modeling/cluster/dbscan.py CHANGED
@@ -561,6 +561,7 @@ class DBSCAN(BaseTransformer):
561
561
  """
562
562
  self._infer_input_output_cols(dataset)
563
563
  super()._check_dataset_type(dataset)
564
+
564
565
  model_trainer = ModelTrainerBuilder.build_fit_transform(
565
566
  estimator=self._sklearn_object,
566
567
  dataset=dataset,
snowflake/ml/modeling/cluster/feature_agglomeration.py CHANGED
@@ -597,6 +597,7 @@ class FeatureAgglomeration(BaseTransformer):
597
597
  """
598
598
  self._infer_input_output_cols(dataset)
599
599
  super()._check_dataset_type(dataset)
600
+
600
601
  model_trainer = ModelTrainerBuilder.build_fit_transform(
601
602
  estimator=self._sklearn_object,
602
603
  dataset=dataset,
snowflake/ml/modeling/cluster/k_means.py CHANGED
@@ -594,6 +594,7 @@ class KMeans(BaseTransformer):
594
594
  """
595
595
  self._infer_input_output_cols(dataset)
596
596
  super()._check_dataset_type(dataset)
597
+
597
598
  model_trainer = ModelTrainerBuilder.build_fit_transform(
598
599
  estimator=self._sklearn_object,
599
600
  dataset=dataset,
snowflake/ml/modeling/cluster/mean_shift.py CHANGED
@@ -566,6 +566,7 @@ class MeanShift(BaseTransformer):
566
566
  """
567
567
  self._infer_input_output_cols(dataset)
568
568
  super()._check_dataset_type(dataset)
569
+
569
570
  model_trainer = ModelTrainerBuilder.build_fit_transform(
570
571
  estimator=self._sklearn_object,
571
572
  dataset=dataset,
snowflake/ml/modeling/cluster/mini_batch_k_means.py CHANGED
@@ -620,6 +620,7 @@ class MiniBatchKMeans(BaseTransformer):
620
620
  """
621
621
  self._infer_input_output_cols(dataset)
622
622
  super()._check_dataset_type(dataset)
623
+
623
624
  model_trainer = ModelTrainerBuilder.build_fit_transform(
624
625
  estimator=self._sklearn_object,
625
626
  dataset=dataset,
snowflake/ml/modeling/cluster/optics.py CHANGED
@@ -634,6 +634,7 @@ class OPTICS(BaseTransformer):
634
634
  """
635
635
  self._infer_input_output_cols(dataset)
636
636
  super()._check_dataset_type(dataset)
637
+
637
638
  model_trainer = ModelTrainerBuilder.build_fit_transform(
638
639
  estimator=self._sklearn_object,
639
640
  dataset=dataset,
snowflake/ml/modeling/cluster/spectral_biclustering.py CHANGED
@@ -570,6 +570,7 @@ class SpectralBiclustering(BaseTransformer):
570
570
  """
571
571
  self._infer_input_output_cols(dataset)
572
572
  super()._check_dataset_type(dataset)
573
+
573
574
  model_trainer = ModelTrainerBuilder.build_fit_transform(
574
575
  estimator=self._sklearn_object,
575
576
  dataset=dataset,
snowflake/ml/modeling/cluster/spectral_clustering.py CHANGED
@@ -630,6 +630,7 @@ class SpectralClustering(BaseTransformer):
630
630
  """
631
631
  self._infer_input_output_cols(dataset)
632
632
  super()._check_dataset_type(dataset)
633
+
633
634
  model_trainer = ModelTrainerBuilder.build_fit_transform(
634
635
  estimator=self._sklearn_object,
635
636
  dataset=dataset,
snowflake/ml/modeling/cluster/spectral_coclustering.py CHANGED
@@ -549,6 +549,7 @@ class SpectralCoclustering(BaseTransformer):
549
549
  """
550
550
  self._infer_input_output_cols(dataset)
551
551
  super()._check_dataset_type(dataset)
552
+
552
553
  model_trainer = ModelTrainerBuilder.build_fit_transform(
553
554
  estimator=self._sklearn_object,
554
555
  dataset=dataset,
snowflake/ml/modeling/compose/column_transformer.py CHANGED
@@ -583,6 +583,7 @@ class ColumnTransformer(BaseTransformer):
583
583
  """
584
584
  self._infer_input_output_cols(dataset)
585
585
  super()._check_dataset_type(dataset)
586
+
586
587
  model_trainer = ModelTrainerBuilder.build_fit_transform(
587
588
  estimator=self._sklearn_object,
588
589
  dataset=dataset,
snowflake/ml/modeling/compose/transformed_target_regressor.py CHANGED
@@ -540,6 +540,7 @@ class TransformedTargetRegressor(BaseTransformer):
540
540
  """
541
541
  self._infer_input_output_cols(dataset)
542
542
  super()._check_dataset_type(dataset)
543
+
543
544
  model_trainer = ModelTrainerBuilder.build_fit_transform(
544
545
  estimator=self._sklearn_object,
545
546
  dataset=dataset,
snowflake/ml/modeling/covariance/elliptic_envelope.py CHANGED
@@ -539,6 +539,7 @@ class EllipticEnvelope(BaseTransformer):
539
539
  """
540
540
  self._infer_input_output_cols(dataset)
541
541
  super()._check_dataset_type(dataset)
542
+
542
543
  model_trainer = ModelTrainerBuilder.build_fit_transform(
543
544
  estimator=self._sklearn_object,
544
545
  dataset=dataset,
snowflake/ml/modeling/covariance/empirical_covariance.py CHANGED
@@ -511,6 +511,7 @@ class EmpiricalCovariance(BaseTransformer):
511
511
  """
512
512
  self._infer_input_output_cols(dataset)
513
513
  super()._check_dataset_type(dataset)
514
+
514
515
  model_trainer = ModelTrainerBuilder.build_fit_transform(
515
516
  estimator=self._sklearn_object,
516
517
  dataset=dataset,
snowflake/ml/modeling/covariance/graphical_lasso.py CHANGED
@@ -559,6 +559,7 @@ class GraphicalLasso(BaseTransformer):
559
559
  """
560
560
  self._infer_input_output_cols(dataset)
561
561
  super()._check_dataset_type(dataset)
562
+
562
563
  model_trainer = ModelTrainerBuilder.build_fit_transform(
563
564
  estimator=self._sklearn_object,
564
565
  dataset=dataset,
snowflake/ml/modeling/covariance/graphical_lasso_cv.py CHANGED
@@ -585,6 +585,7 @@ class GraphicalLassoCV(BaseTransformer):
585
585
  """
586
586
  self._infer_input_output_cols(dataset)
587
587
  super()._check_dataset_type(dataset)
588
+
588
589
  model_trainer = ModelTrainerBuilder.build_fit_transform(
589
590
  estimator=self._sklearn_object,
590
591
  dataset=dataset,
snowflake/ml/modeling/covariance/ledoit_wolf.py CHANGED
@@ -518,6 +518,7 @@ class LedoitWolf(BaseTransformer):
518
518
  """
519
519
  self._infer_input_output_cols(dataset)
520
520
  super()._check_dataset_type(dataset)
521
+
521
522
  model_trainer = ModelTrainerBuilder.build_fit_transform(
522
523
  estimator=self._sklearn_object,
523
524
  dataset=dataset,
snowflake/ml/modeling/covariance/min_cov_det.py CHANGED
@@ -530,6 +530,7 @@ class MinCovDet(BaseTransformer):
530
530
  """
531
531
  self._infer_input_output_cols(dataset)
532
532
  super()._check_dataset_type(dataset)
533
+
533
534
  model_trainer = ModelTrainerBuilder.build_fit_transform(
534
535
  estimator=self._sklearn_object,
535
536
  dataset=dataset,
snowflake/ml/modeling/covariance/oas.py CHANGED
@@ -511,6 +511,7 @@ class OAS(BaseTransformer):
511
511
  """
512
512
  self._infer_input_output_cols(dataset)
513
513
  super()._check_dataset_type(dataset)
514
+
514
515
  model_trainer = ModelTrainerBuilder.build_fit_transform(
515
516
  estimator=self._sklearn_object,
516
517
  dataset=dataset,
snowflake/ml/modeling/covariance/shrunk_covariance.py CHANGED
@@ -517,6 +517,7 @@ class ShrunkCovariance(BaseTransformer):
517
517
  """
518
518
  self._infer_input_output_cols(dataset)
519
519
  super()._check_dataset_type(dataset)
520
+
520
521
  model_trainer = ModelTrainerBuilder.build_fit_transform(
521
522
  estimator=self._sklearn_object,
522
523
  dataset=dataset,
snowflake/ml/modeling/decomposition/dictionary_learning.py CHANGED
@@ -627,6 +627,7 @@ class DictionaryLearning(BaseTransformer):
627
627
  """
628
628
  self._infer_input_output_cols(dataset)
629
629
  super()._check_dataset_type(dataset)
630
+
630
631
  model_trainer = ModelTrainerBuilder.build_fit_transform(
631
632
  estimator=self._sklearn_object,
632
633
  dataset=dataset,
snowflake/ml/modeling/decomposition/factor_analysis.py CHANGED
@@ -564,6 +564,7 @@ class FactorAnalysis(BaseTransformer):
564
564
  """
565
565
  self._infer_input_output_cols(dataset)
566
566
  super()._check_dataset_type(dataset)
567
+
567
568
  model_trainer = ModelTrainerBuilder.build_fit_transform(
568
569
  estimator=self._sklearn_object,
569
570
  dataset=dataset,
snowflake/ml/modeling/decomposition/fast_ica.py CHANGED
@@ -582,6 +582,7 @@ class FastICA(BaseTransformer):
582
582
  """
583
583
  self._infer_input_output_cols(dataset)
584
584
  super()._check_dataset_type(dataset)
585
+
585
586
  model_trainer = ModelTrainerBuilder.build_fit_transform(
586
587
  estimator=self._sklearn_object,
587
588
  dataset=dataset,
snowflake/ml/modeling/decomposition/incremental_pca.py CHANGED
@@ -534,6 +534,7 @@ class IncrementalPCA(BaseTransformer):
534
534
  """
535
535
  self._infer_input_output_cols(dataset)
536
536
  super()._check_dataset_type(dataset)
537
+
537
538
  model_trainer = ModelTrainerBuilder.build_fit_transform(
538
539
  estimator=self._sklearn_object,
539
540
  dataset=dataset,
snowflake/ml/modeling/decomposition/kernel_pca.py CHANGED
@@ -630,6 +630,7 @@ class KernelPCA(BaseTransformer):
630
630
  """
631
631
  self._infer_input_output_cols(dataset)
632
632
  super()._check_dataset_type(dataset)
633
+
633
634
  model_trainer = ModelTrainerBuilder.build_fit_transform(
634
635
  estimator=self._sklearn_object,
635
636
  dataset=dataset,
snowflake/ml/modeling/decomposition/mini_batch_dictionary_learning.py CHANGED
@@ -652,6 +652,7 @@ class MiniBatchDictionaryLearning(BaseTransformer):
652
652
  """
653
653
  self._infer_input_output_cols(dataset)
654
654
  super()._check_dataset_type(dataset)
655
+
655
656
  model_trainer = ModelTrainerBuilder.build_fit_transform(
656
657
  estimator=self._sklearn_object,
657
658
  dataset=dataset,
snowflake/ml/modeling/decomposition/mini_batch_sparse_pca.py CHANGED
@@ -597,6 +597,7 @@ class MiniBatchSparsePCA(BaseTransformer):
597
597
  """
598
598
  self._infer_input_output_cols(dataset)
599
599
  super()._check_dataset_type(dataset)
600
+
600
601
  model_trainer = ModelTrainerBuilder.build_fit_transform(
601
602
  estimator=self._sklearn_object,
602
603
  dataset=dataset,
snowflake/ml/modeling/decomposition/pca.py CHANGED
@@ -599,6 +599,7 @@ class PCA(BaseTransformer):
599
599
  """
600
600
  self._infer_input_output_cols(dataset)
601
601
  super()._check_dataset_type(dataset)
602
+
602
603
  model_trainer = ModelTrainerBuilder.build_fit_transform(
603
604
  estimator=self._sklearn_object,
604
605
  dataset=dataset,
snowflake/ml/modeling/decomposition/sparse_pca.py CHANGED
@@ -572,6 +572,7 @@ class SparsePCA(BaseTransformer):
572
572
  """
573
573
  self._infer_input_output_cols(dataset)
574
574
  super()._check_dataset_type(dataset)
575
+
575
576
  model_trainer = ModelTrainerBuilder.build_fit_transform(
576
577
  estimator=self._sklearn_object,
577
578
  dataset=dataset,
snowflake/ml/modeling/decomposition/truncated_svd.py CHANGED
@@ -553,6 +553,7 @@ class TruncatedSVD(BaseTransformer):
553
553
  """
554
554
  self._infer_input_output_cols(dataset)
555
555
  super()._check_dataset_type(dataset)
556
+
556
557
  model_trainer = ModelTrainerBuilder.build_fit_transform(
557
558
  estimator=self._sklearn_object,
558
559
  dataset=dataset,
snowflake/ml/modeling/discriminant_analysis/linear_discriminant_analysis.py CHANGED
@@ -570,6 +570,7 @@ class LinearDiscriminantAnalysis(BaseTransformer):
570
570
  """
571
571
  self._infer_input_output_cols(dataset)
572
572
  super()._check_dataset_type(dataset)
573
+
573
574
  model_trainer = ModelTrainerBuilder.build_fit_transform(
574
575
  estimator=self._sklearn_object,
575
576
  dataset=dataset,
snowflake/ml/modeling/discriminant_analysis/quadratic_discriminant_analysis.py CHANGED
@@ -528,6 +528,7 @@ class QuadraticDiscriminantAnalysis(BaseTransformer):
528
528
  """
529
529
  self._infer_input_output_cols(dataset)
530
530
  super()._check_dataset_type(dataset)
531
+
531
532
  model_trainer = ModelTrainerBuilder.build_fit_transform(
532
533
  estimator=self._sklearn_object,
533
534
  dataset=dataset,
snowflake/ml/modeling/ensemble/ada_boost_classifier.py CHANGED
@@ -553,6 +553,7 @@ class AdaBoostClassifier(BaseTransformer):
553
553
  """
554
554
  self._infer_input_output_cols(dataset)
555
555
  super()._check_dataset_type(dataset)
556
+
556
557
  model_trainer = ModelTrainerBuilder.build_fit_transform(
557
558
  estimator=self._sklearn_object,
558
559
  dataset=dataset,
snowflake/ml/modeling/ensemble/ada_boost_regressor.py CHANGED
@@ -550,6 +550,7 @@ class AdaBoostRegressor(BaseTransformer):
550
550
  """
551
551
  self._infer_input_output_cols(dataset)
552
552
  super()._check_dataset_type(dataset)
553
+
553
554
  model_trainer = ModelTrainerBuilder.build_fit_transform(
554
555
  estimator=self._sklearn_object,
555
556
  dataset=dataset,
snowflake/ml/modeling/ensemble/bagging_classifier.py CHANGED
@@ -585,6 +585,7 @@ class BaggingClassifier(BaseTransformer):
585
585
  """
586
586
  self._infer_input_output_cols(dataset)
587
587
  super()._check_dataset_type(dataset)
588
+
588
589
  model_trainer = ModelTrainerBuilder.build_fit_transform(
589
590
  estimator=self._sklearn_object,
590
591
  dataset=dataset,
snowflake/ml/modeling/ensemble/bagging_regressor.py CHANGED
@@ -585,6 +585,7 @@ class BaggingRegressor(BaseTransformer):
585
585
  """
586
586
  self._infer_input_output_cols(dataset)
587
587
  super()._check_dataset_type(dataset)
588
+
588
589
  model_trainer = ModelTrainerBuilder.build_fit_transform(
589
590
  estimator=self._sklearn_object,
590
591
  dataset=dataset,
snowflake/ml/modeling/ensemble/extra_trees_classifier.py CHANGED
@@ -688,6 +688,7 @@ class ExtraTreesClassifier(BaseTransformer):
688
688
  """
689
689
  self._infer_input_output_cols(dataset)
690
690
  super()._check_dataset_type(dataset)
691
+
691
692
  model_trainer = ModelTrainerBuilder.build_fit_transform(
692
693
  estimator=self._sklearn_object,
693
694
  dataset=dataset,
snowflake/ml/modeling/ensemble/extra_trees_regressor.py CHANGED
@@ -667,6 +667,7 @@ class ExtraTreesRegressor(BaseTransformer):
667
667
  """
668
668
  self._infer_input_output_cols(dataset)
669
669
  super()._check_dataset_type(dataset)
670
+
670
671
  model_trainer = ModelTrainerBuilder.build_fit_transform(
671
672
  estimator=self._sklearn_object,
672
673
  dataset=dataset,
snowflake/ml/modeling/ensemble/gradient_boosting_classifier.py CHANGED
@@ -700,6 +700,7 @@ class GradientBoostingClassifier(BaseTransformer):
700
700
  """
701
701
  self._infer_input_output_cols(dataset)
702
702
  super()._check_dataset_type(dataset)
703
+
703
704
  model_trainer = ModelTrainerBuilder.build_fit_transform(
704
705
  estimator=self._sklearn_object,
705
706
  dataset=dataset,
snowflake/ml/modeling/ensemble/gradient_boosting_regressor.py CHANGED
@@ -709,6 +709,7 @@ class GradientBoostingRegressor(BaseTransformer):
709
709
  """
710
710
  self._infer_input_output_cols(dataset)
711
711
  super()._check_dataset_type(dataset)
712
+
712
713
  model_trainer = ModelTrainerBuilder.build_fit_transform(
713
714
  estimator=self._sklearn_object,
714
715
  dataset=dataset,
snowflake/ml/modeling/ensemble/hist_gradient_boosting_classifier.py CHANGED
@@ -681,6 +681,7 @@ class HistGradientBoostingClassifier(BaseTransformer):
681
681
  """
682
682
  self._infer_input_output_cols(dataset)
683
683
  super()._check_dataset_type(dataset)
684
+
684
685
  model_trainer = ModelTrainerBuilder.build_fit_transform(
685
686
  estimator=self._sklearn_object,
686
687
  dataset=dataset,
snowflake/ml/modeling/ensemble/hist_gradient_boosting_regressor.py CHANGED
@@ -672,6 +672,7 @@ class HistGradientBoostingRegressor(BaseTransformer):
672
672
  """
673
673
  self._infer_input_output_cols(dataset)
674
674
  super()._check_dataset_type(dataset)
675
+
675
676
  model_trainer = ModelTrainerBuilder.build_fit_transform(
676
677
  estimator=self._sklearn_object,
677
678
  dataset=dataset,
snowflake/ml/modeling/ensemble/isolation_forest.py CHANGED
@@ -576,6 +576,7 @@ class IsolationForest(BaseTransformer):
576
576
  """
577
577
  self._infer_input_output_cols(dataset)
578
578
  super()._check_dataset_type(dataset)
579
+
579
580
  model_trainer = ModelTrainerBuilder.build_fit_transform(
580
581
  estimator=self._sklearn_object,
581
582
  dataset=dataset,
snowflake/ml/modeling/ensemble/random_forest_classifier.py CHANGED
@@ -684,6 +684,7 @@ class RandomForestClassifier(BaseTransformer):
684
684
  """
685
685
  self._infer_input_output_cols(dataset)
686
686
  super()._check_dataset_type(dataset)
687
+
687
688
  model_trainer = ModelTrainerBuilder.build_fit_transform(
688
689
  estimator=self._sklearn_object,
689
690
  dataset=dataset,
snowflake/ml/modeling/ensemble/random_forest_regressor.py CHANGED
@@ -663,6 +663,7 @@ class RandomForestRegressor(BaseTransformer):
663
663
  """
664
664
  self._infer_input_output_cols(dataset)
665
665
  super()._check_dataset_type(dataset)
666
+
666
667
  model_trainer = ModelTrainerBuilder.build_fit_transform(
667
668
  estimator=self._sklearn_object,
668
669
  dataset=dataset,
snowflake/ml/modeling/ensemble/stacking_regressor.py CHANGED
@@ -568,6 +568,7 @@ class StackingRegressor(BaseTransformer):
568
568
  """
569
569
  self._infer_input_output_cols(dataset)
570
570
  super()._check_dataset_type(dataset)
571
+
571
572
  model_trainer = ModelTrainerBuilder.build_fit_transform(
572
573
  estimator=self._sklearn_object,
573
574
  dataset=dataset,
snowflake/ml/modeling/ensemble/voting_classifier.py CHANGED
@@ -550,6 +550,7 @@ class VotingClassifier(BaseTransformer):
550
550
  """
551
551
  self._infer_input_output_cols(dataset)
552
552
  super()._check_dataset_type(dataset)
553
+
553
554
  model_trainer = ModelTrainerBuilder.build_fit_transform(
554
555
  estimator=self._sklearn_object,
555
556
  dataset=dataset,
snowflake/ml/modeling/ensemble/voting_regressor.py CHANGED
@@ -532,6 +532,7 @@ class VotingRegressor(BaseTransformer):
532
532
  """
533
533
  self._infer_input_output_cols(dataset)
534
534
  super()._check_dataset_type(dataset)
535
+
535
536
  model_trainer = ModelTrainerBuilder.build_fit_transform(
536
537
  estimator=self._sklearn_object,
537
538
  dataset=dataset,
snowflake/ml/modeling/feature_selection/generic_univariate_select.py CHANGED
@@ -520,6 +520,7 @@ class GenericUnivariateSelect(BaseTransformer):
520
520
  """
521
521
  self._infer_input_output_cols(dataset)
522
522
  super()._check_dataset_type(dataset)
523
+
523
524
  model_trainer = ModelTrainerBuilder.build_fit_transform(
524
525
  estimator=self._sklearn_object,
525
526
  dataset=dataset,
snowflake/ml/modeling/feature_selection/select_fdr.py CHANGED
@@ -516,6 +516,7 @@ class SelectFdr(BaseTransformer):
516
516
  """
517
517
  self._infer_input_output_cols(dataset)
518
518
  super()._check_dataset_type(dataset)
519
+
519
520
  model_trainer = ModelTrainerBuilder.build_fit_transform(
520
521
  estimator=self._sklearn_object,
521
522
  dataset=dataset,
snowflake/ml/modeling/feature_selection/select_fpr.py CHANGED
@@ -516,6 +516,7 @@ class SelectFpr(BaseTransformer):
516
516
  """
517
517
  self._infer_input_output_cols(dataset)
518
518
  super()._check_dataset_type(dataset)
519
+
519
520
  model_trainer = ModelTrainerBuilder.build_fit_transform(
520
521
  estimator=self._sklearn_object,
521
522
  dataset=dataset,
snowflake/ml/modeling/feature_selection/select_fwe.py CHANGED
@@ -516,6 +516,7 @@ class SelectFwe(BaseTransformer):
516
516
  """
517
517
  self._infer_input_output_cols(dataset)
518
518
  super()._check_dataset_type(dataset)
519
+
519
520
  model_trainer = ModelTrainerBuilder.build_fit_transform(
520
521
  estimator=self._sklearn_object,
521
522
  dataset=dataset,
snowflake/ml/modeling/feature_selection/select_k_best.py CHANGED
@@ -517,6 +517,7 @@ class SelectKBest(BaseTransformer):
517
517
  """
518
518
  self._infer_input_output_cols(dataset)
519
519
  super()._check_dataset_type(dataset)
520
+
520
521
  model_trainer = ModelTrainerBuilder.build_fit_transform(
521
522
  estimator=self._sklearn_object,
522
523
  dataset=dataset,
snowflake/ml/modeling/feature_selection/select_percentile.py CHANGED
@@ -516,6 +516,7 @@ class SelectPercentile(BaseTransformer):
516
516
  """
517
517
  self._infer_input_output_cols(dataset)
518
518
  super()._check_dataset_type(dataset)
519
+
519
520
  model_trainer = ModelTrainerBuilder.build_fit_transform(
520
521
  estimator=self._sklearn_object,
521
522
  dataset=dataset,
snowflake/ml/modeling/feature_selection/sequential_feature_selector.py CHANGED
@@ -576,6 +576,7 @@ class SequentialFeatureSelector(BaseTransformer):
576
576
  """
577
577
  self._infer_input_output_cols(dataset)
578
578
  super()._check_dataset_type(dataset)
579
+
579
580
  model_trainer = ModelTrainerBuilder.build_fit_transform(
580
581
  estimator=self._sklearn_object,
581
582
  dataset=dataset,
snowflake/ml/modeling/feature_selection/variance_threshold.py CHANGED
@@ -509,6 +509,7 @@ class VarianceThreshold(BaseTransformer):
509
509
  """
510
510
  self._infer_input_output_cols(dataset)
511
511
  super()._check_dataset_type(dataset)
512
+
512
513
  model_trainer = ModelTrainerBuilder.build_fit_transform(
513
514
  estimator=self._sklearn_object,
514
515
  dataset=dataset,
snowflake/ml/modeling/gaussian_process/gaussian_process_classifier.py CHANGED
@@ -600,6 +600,7 @@ class GaussianProcessClassifier(BaseTransformer):
600
600
  """
601
601
  self._infer_input_output_cols(dataset)
602
602
  super()._check_dataset_type(dataset)
603
+
603
604
  model_trainer = ModelTrainerBuilder.build_fit_transform(
604
605
  estimator=self._sklearn_object,
605
606
  dataset=dataset,
snowflake/ml/modeling/gaussian_process/gaussian_process_regressor.py CHANGED
@@ -591,6 +591,7 @@ class GaussianProcessRegressor(BaseTransformer):
591
591
  """
592
592
  self._infer_input_output_cols(dataset)
593
593
  super()._check_dataset_type(dataset)
594
+
594
595
  model_trainer = ModelTrainerBuilder.build_fit_transform(
595
596
  estimator=self._sklearn_object,
596
597
  dataset=dataset,
snowflake/ml/modeling/impute/iterative_imputer.py CHANGED
@@ -637,6 +637,7 @@ class IterativeImputer(BaseTransformer):
637
637
  """
638
638
  self._infer_input_output_cols(dataset)
639
639
  super()._check_dataset_type(dataset)
640
+
640
641
  model_trainer = ModelTrainerBuilder.build_fit_transform(
641
642
  estimator=self._sklearn_object,
642
643
  dataset=dataset,
snowflake/ml/modeling/impute/knn_imputer.py CHANGED
@@ -563,6 +563,7 @@ class KNNImputer(BaseTransformer):
563
563
  """
564
564
  self._infer_input_output_cols(dataset)
565
565
  super()._check_dataset_type(dataset)
566
+
566
567
  model_trainer = ModelTrainerBuilder.build_fit_transform(
567
568
  estimator=self._sklearn_object,
568
569
  dataset=dataset,
snowflake/ml/modeling/impute/missing_indicator.py CHANGED
@@ -537,6 +537,7 @@ class MissingIndicator(BaseTransformer):
537
537
  """
538
538
  self._infer_input_output_cols(dataset)
539
539
  super()._check_dataset_type(dataset)
540
+
540
541
  model_trainer = ModelTrainerBuilder.build_fit_transform(
541
542
  estimator=self._sklearn_object,
542
543
  dataset=dataset,
snowflake/ml/modeling/kernel_approximation/additive_chi2_sampler.py CHANGED
@@ -512,6 +512,7 @@ class AdditiveChi2Sampler(BaseTransformer):
512
512
  """
513
513
  self._infer_input_output_cols(dataset)
514
514
  super()._check_dataset_type(dataset)
515
+
515
516
  model_trainer = ModelTrainerBuilder.build_fit_transform(
516
517
  estimator=self._sklearn_object,
517
518
  dataset=dataset,
snowflake/ml/modeling/kernel_approximation/nystroem.py CHANGED
@@ -560,6 +560,7 @@ class Nystroem(BaseTransformer):
560
560
  """
561
561
  self._infer_input_output_cols(dataset)
562
562
  super()._check_dataset_type(dataset)
563
+
563
564
  model_trainer = ModelTrainerBuilder.build_fit_transform(
564
565
  estimator=self._sklearn_object,
565
566
  dataset=dataset,
snowflake/ml/modeling/kernel_approximation/polynomial_count_sketch.py CHANGED
@@ -536,6 +536,7 @@ class PolynomialCountSketch(BaseTransformer):
536
536
  """
537
537
  self._infer_input_output_cols(dataset)
538
538
  super()._check_dataset_type(dataset)
539
+
539
540
  model_trainer = ModelTrainerBuilder.build_fit_transform(
540
541
  estimator=self._sklearn_object,
541
542
  dataset=dataset,
snowflake/ml/modeling/kernel_approximation/rbf_sampler.py CHANGED
@@ -523,6 +523,7 @@ class RBFSampler(BaseTransformer):
523
523
  """
524
524
  self._infer_input_output_cols(dataset)
525
525
  super()._check_dataset_type(dataset)
526
+
526
527
  model_trainer = ModelTrainerBuilder.build_fit_transform(
527
528
  estimator=self._sklearn_object,
528
529
  dataset=dataset,
snowflake/ml/modeling/kernel_approximation/skewed_chi2_sampler.py CHANGED
@@ -521,6 +521,7 @@ class SkewedChi2Sampler(BaseTransformer):
521
521
  """
522
522
  self._infer_input_output_cols(dataset)
523
523
  super()._check_dataset_type(dataset)
524
+
524
525
  model_trainer = ModelTrainerBuilder.build_fit_transform(
525
526
  estimator=self._sklearn_object,
526
527
  dataset=dataset,
snowflake/ml/modeling/kernel_ridge/kernel_ridge.py CHANGED
@@ -553,6 +553,7 @@ class KernelRidge(BaseTransformer):
553
553
  """
554
554
  self._infer_input_output_cols(dataset)
555
555
  super()._check_dataset_type(dataset)
556
+
556
557
  model_trainer = ModelTrainerBuilder.build_fit_transform(
557
558
  estimator=self._sklearn_object,
558
559
  dataset=dataset,
snowflake/ml/modeling/lightgbm/lgbm_classifier.py CHANGED
@@ -542,6 +542,7 @@ class LGBMClassifier(BaseTransformer):
542
542
  """
543
543
  self._infer_input_output_cols(dataset)
544
544
  super()._check_dataset_type(dataset)
545
+
545
546
  model_trainer = ModelTrainerBuilder.build_fit_transform(
546
547
  estimator=self._sklearn_object,
547
548
  dataset=dataset,
snowflake/ml/modeling/lightgbm/lgbm_regressor.py CHANGED
@@ -542,6 +542,7 @@ class LGBMRegressor(BaseTransformer):
542
542
  """
543
543
  self._infer_input_output_cols(dataset)
544
544
  super()._check_dataset_type(dataset)
545
+
545
546
  model_trainer = ModelTrainerBuilder.build_fit_transform(
546
547
  estimator=self._sklearn_object,
547
548
  dataset=dataset,