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

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

@@ -13,12 +13,12 @@ from snowflake.ml._internal.exceptions import (
     exceptions,
     modeling_error_messages,
 )
-from snowflake.ml._internal.utils import pkg_version_utils
+from snowflake.ml._internal.utils import pkg_version_utils, temp_file_utils
 from snowflake.ml._internal.utils.query_result_checker import ResultValidator
 from snowflake.ml._internal.utils.snowpark_dataframe_utils import (
     cast_snowpark_dataframe,
 )
-from snowflake.ml._internal.utils.temp_file_utils import get_temp_file_path
+from snowflake.ml.modeling._internal import estimator_utils
 from snowflake.ml.modeling._internal.model_specifications import (
     ModelSpecifications,
     ModelSpecificationsBuilder,
@@ -306,8 +306,6 @@ class XGBoostExternalMemoryTrainer(SnowparkModelTrainer):
         )  # type: ignore[misc]
         def fit_wrapper_sproc(
             session: Session,
-            stage_transform_file_name: str,
-            stage_result_file_name: str,
             dataset_stage_name: str,
             batch_size: int,
             input_cols: List[str],
@@ -320,9 +318,13 @@ class XGBoostExternalMemoryTrainer(SnowparkModelTrainer):
 
             import cloudpickle as cp
 
-            local_transform_file_name = get_temp_file_path()
+            local_transform_file_name = temp_file_utils.get_temp_file_path()
 
-            session.file.get(stage_transform_file_name, local_transform_file_name, statement_params=statement_params)
+            session.file.get(
+                stage_location=dataset_stage_name,
+                target_directory=local_transform_file_name,
+                statement_params=statement_params,
+            )
 
             local_transform_file_path = os.path.join(
                 local_transform_file_name, os.listdir(local_transform_file_name)[0]
@@ -345,13 +347,13 @@ class XGBoostExternalMemoryTrainer(SnowparkModelTrainer):
                 sample_weight_col=sample_weight_col,
             )
 
-            local_result_file_name = get_temp_file_path()
+            local_result_file_name = temp_file_utils.get_temp_file_path()
             with open(local_result_file_name, mode="w+b") as local_result_file_obj:
                 cp.dump(estimator, local_result_file_obj)
 
             session.file.put(
-                local_result_file_name,
-                stage_result_file_name,
+                local_file_name=local_result_file_name,
+                stage_location=dataset_stage_name,
                 auto_compress=False,
                 overwrite=True,
                 statement_params=statement_params,
@@ -394,11 +396,6 @@ class XGBoostExternalMemoryTrainer(SnowparkModelTrainer):
             SnowflakeMLException: For known types of user and system errors.
             e: For every unexpected exception from SnowflakeClient.
         """
-        temp_stage_name = self._create_temp_stage()
-        (stage_transform_file_name, stage_result_file_name) = self._upload_model_to_stage(stage_name=temp_stage_name)
-        data_file_paths = self._write_training_data_to_stage(dataset_stage_name=temp_stage_name)
-
-        # Call fit sproc
         statement_params = telemetry.get_function_usage_statement_params(
             project=_PROJECT,
             subproject=self._subproject,
@@ -406,7 +403,16 @@ class XGBoostExternalMemoryTrainer(SnowparkModelTrainer):
             api_calls=[Session.call],
             custom_tags=None,
         )
+        temp_stage_name = estimator_utils.create_temp_stage(self.session)
+        estimator_utils.upload_model_to_stage(
+            stage_name=temp_stage_name,
+            estimator=self.estimator,
+            session=self.session,
+            statement_params=statement_params,
+        )
+        data_file_paths = self._write_training_data_to_stage(dataset_stage_name=temp_stage_name)
 
+        # Call fit sproc
         model_spec = ModelSpecificationsBuilder.build(model=self.estimator)
         fit_wrapper = self._get_xgb_external_memory_fit_wrapper_sproc(
             model_spec=model_spec,
@@ -418,8 +424,6 @@ class XGBoostExternalMemoryTrainer(SnowparkModelTrainer):
         try:
             sproc_export_file_name = fit_wrapper(
                 self.session,
-                stage_transform_file_name,
-                stage_result_file_name,
                 temp_stage_name,
                 self._batch_size,
                 self.input_cols,
@@ -440,7 +444,7 @@ class XGBoostExternalMemoryTrainer(SnowparkModelTrainer):
             sproc_export_file_name = fields[0]
 
         return self._fetch_model_from_stage(
-            dir_path=stage_result_file_name,
+            dir_path=temp_stage_name,
             file_name=sproc_export_file_name,
             statement_params=statement_params,
         )

snowflake/ml/modeling/calibration/calibrated_classifier_cv.py

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

snowflake/ml/modeling/cluster/affinity_propagation.py

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

snowflake/ml/modeling/cluster/agglomerative_clustering.py

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

snowflake/ml/modeling/cluster/birch.py

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

snowflake/ml/modeling/cluster/bisecting_k_means.py

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

snowflake/ml/modeling/cluster/dbscan.py

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

snowflake/ml/modeling/cluster/feature_agglomeration.py

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

snowflake/ml/modeling/cluster/k_means.py

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

snowflake/ml/modeling/cluster/mean_shift.py

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

snowflake/ml/modeling/cluster/mini_batch_k_means.py

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

snowflake/ml/modeling/cluster/optics.py

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

snowflake/ml/modeling/cluster/spectral_biclustering.py

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

snowflake/ml/modeling/cluster/spectral_clustering.py

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

snowflake/ml/modeling/cluster/spectral_coclustering.py

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

snowflake/ml/modeling/compose/column_transformer.py

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

snowflake/ml/modeling/compose/transformed_target_regressor.py

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

snowflake/ml/modeling/covariance/elliptic_envelope.py

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

snowflake/ml/modeling/covariance/empirical_covariance.py

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

snowflake/ml/modeling/covariance/graphical_lasso.py

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

snowflake/ml/modeling/covariance/graphical_lasso_cv.py

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

snowflake/ml/modeling/covariance/ledoit_wolf.py

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

snowflake/ml/modeling/covariance/min_cov_det.py

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

snowflake/ml/modeling/covariance/oas.py

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

snowflake/ml/modeling/covariance/shrunk_covariance.py

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

snowflake/ml/modeling/decomposition/dictionary_learning.py

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

snowflake/ml/modeling/decomposition/factor_analysis.py

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

snowflake/ml/modeling/decomposition/fast_ica.py

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

snowflake/ml/modeling/decomposition/incremental_pca.py

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

snowflake/ml/modeling/decomposition/kernel_pca.py

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

snowflake/ml/modeling/decomposition/mini_batch_dictionary_learning.py

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

snowflake/ml/modeling/decomposition/mini_batch_sparse_pca.py

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

snowflake/ml/modeling/decomposition/pca.py

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

snowflake/ml/modeling/decomposition/sparse_pca.py

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

snowflake/ml/modeling/decomposition/truncated_svd.py

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

snowflake/ml/modeling/discriminant_analysis/linear_discriminant_analysis.py

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

snowflake/ml/modeling/discriminant_analysis/quadratic_discriminant_analysis.py

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

snowflake/ml/modeling/ensemble/ada_boost_classifier.py

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

snowflake/ml/modeling/ensemble/ada_boost_regressor.py

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

snowflake/ml/modeling/ensemble/bagging_classifier.py

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

snowflake/ml/modeling/ensemble/bagging_regressor.py

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

snowflake/ml/modeling/ensemble/extra_trees_classifier.py

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

snowflake/ml/modeling/ensemble/extra_trees_regressor.py

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

snowflake/ml/modeling/ensemble/gradient_boosting_classifier.py

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

snowflake/ml/modeling/ensemble/gradient_boosting_regressor.py

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

snowflake/ml/modeling/ensemble/hist_gradient_boosting_classifier.py

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

snowflake/ml/modeling/ensemble/hist_gradient_boosting_regressor.py

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

snowflake/ml/modeling/ensemble/isolation_forest.py

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

snowflake/ml/modeling/ensemble/random_forest_classifier.py

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

snowflake/ml/modeling/ensemble/random_forest_regressor.py

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