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

snowflake/ml/modeling/ensemble/stacking_regressor.py

@@ -284,7 +284,7 @@ class StackingRegressor(BaseTransformer):
                         inspect.currentframe(), StackingRegressor.__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/voting_classifier.py

@@ -266,7 +266,7 @@ class VotingClassifier(BaseTransformer):
                         inspect.currentframe(), VotingClassifier.__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/voting_regressor.py

@@ -248,7 +248,7 @@ class VotingRegressor(BaseTransformer):
                         inspect.currentframe(), VotingRegressor.__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/feature_selection/generic_univariate_select.py

@@ -238,7 +238,7 @@ class GenericUnivariateSelect(BaseTransformer):
                         inspect.currentframe(), GenericUnivariateSelect.__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/feature_selection/select_fdr.py

@@ -234,7 +234,7 @@ class SelectFdr(BaseTransformer):
                         inspect.currentframe(), SelectFdr.__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/feature_selection/select_fpr.py

@@ -234,7 +234,7 @@ class SelectFpr(BaseTransformer):
                         inspect.currentframe(), SelectFpr.__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/feature_selection/select_fwe.py

@@ -234,7 +234,7 @@ class SelectFwe(BaseTransformer):
                         inspect.currentframe(), SelectFwe.__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/feature_selection/select_k_best.py

@@ -235,7 +235,7 @@ class SelectKBest(BaseTransformer):
                         inspect.currentframe(), SelectKBest.__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/feature_selection/select_percentile.py

@@ -234,7 +234,7 @@ class SelectPercentile(BaseTransformer):
                         inspect.currentframe(), SelectPercentile.__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/feature_selection/sequential_feature_selector.py

@@ -292,7 +292,7 @@ class SequentialFeatureSelector(BaseTransformer):
                         inspect.currentframe(), SequentialFeatureSelector.__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/feature_selection/variance_threshold.py

@@ -225,7 +225,7 @@ class VarianceThreshold(BaseTransformer):
                         inspect.currentframe(), VarianceThreshold.__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/framework/base.py

@@ -16,7 +16,7 @@ from snowflake.ml._internal.exceptions import (
     exceptions,
     modeling_error_messages,
 )
-from snowflake.ml._internal.lineage import data_source, lineage_utils
+from snowflake.ml._internal.lineage import lineage_utils
 from snowflake.ml._internal.utils import identifier, parallelize
 from snowflake.ml.modeling.framework import _utils
 from snowflake.snowpark import functions as F
@@ -386,7 +386,6 @@ class BaseEstimator(Base):
         self.file_names = file_names
         self.custom_states = custom_states
         self.sample_weight_col = sample_weight_col
-        self._data_sources: Optional[List[data_source.DataSource]] = None
 
         self.start_time = datetime.now().strftime(_utils.DATETIME_FORMAT)[:-3]
 
@@ -421,18 +420,14 @@ class BaseEstimator(Base):
         """
         return []
 
-    def _get_data_sources(self) -> Optional[List[data_source.DataSource]]:
-        return self._data_sources
-
     @telemetry.send_api_usage_telemetry(
         project=PROJECT,
         subproject=SUBPROJECT,
     )
     def fit(self, dataset: Union[snowpark.DataFrame, pd.DataFrame]) -> "BaseEstimator":
         """Runs universal logics for all fit implementations."""
-        self._data_sources = getattr(dataset, lineage_utils.DATA_SOURCES_ATTR, None)
-        if self._data_sources:
-            assert all(isinstance(ds, data_source.DataSource) for ds in self._data_sources)
+        data_sources = lineage_utils.get_data_sources(dataset)
+        lineage_utils.set_data_sources(self, data_sources)
         return self._fit(dataset)
 
     @abstractmethod

snowflake/ml/modeling/gaussian_process/gaussian_process_classifier.py

@@ -320,7 +320,7 @@ class GaussianProcessClassifier(BaseTransformer):
                         inspect.currentframe(), GaussianProcessClassifier.__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/gaussian_process/gaussian_process_regressor.py

@@ -311,7 +311,7 @@ class GaussianProcessRegressor(BaseTransformer):
                         inspect.currentframe(), GaussianProcessRegressor.__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/impute/iterative_imputer.py

@@ -353,7 +353,7 @@ class IterativeImputer(BaseTransformer):
                         inspect.currentframe(), IterativeImputer.__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/impute/knn_imputer.py

@@ -279,7 +279,7 @@ class KNNImputer(BaseTransformer):
                         inspect.currentframe(), KNNImputer.__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/impute/missing_indicator.py

@@ -253,7 +253,7 @@ class MissingIndicator(BaseTransformer):
                         inspect.currentframe(), MissingIndicator.__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/impute/simple_imputer.py

@@ -102,10 +102,14 @@ class SimpleImputer(base.BaseTransformer):
             For string or object data types, `fill_value` must be a string. If `None`, `fill_value` will be 0 when
             imputing numerical data and `missing_value` for strings and object data types.
         input_cols: Optional[Union[str, List[str]]]
-            Columns to use as inputs during fit and transform.
+            The name(s) of one or more columns in the input DataFrame containing feature(s) to be imputed. Input
+            columns must be specified before fit with this argument or after initialization with the
+            `set_input_cols` method. This argument is optional for API consistency.
         output_cols: Optional[Union[str, List[str]]]
-            A string or list of strings representing column names that will store the output of transform operation.
-            The length of `output_cols` must equal the length of `input_cols`.
+            The name(s) to assign output columns in the output DataFrame. The number of
+            output columns specified must equal the number of input columns. Output columns must be specified before
+            transform with this argument or after initialization with the `set_output_cols` method. This argument is
+            optional for API consistency.
         passthrough_cols: A string or a list of strings indicating column names to be excluded from any
             operations (such as train, transform, or inference). These specified column(s)
             will remain untouched throughout the process. This option is helpful in scenarios
@@ -230,7 +234,7 @@ class SimpleImputer(base.BaseTransformer):
 
         return input_col_datatypes
 
-    def fit(self, dataset: Union[snowpark.DataFrame, pd.DataFrame]) -> "SimpleImputer":
+    def _fit(self, dataset: Union[snowpark.DataFrame, pd.DataFrame]) -> "SimpleImputer":
         if isinstance(dataset, snowpark.DataFrame):
             return self._fit_snowpark(dataset)
         else:

snowflake/ml/modeling/kernel_approximation/additive_chi2_sampler.py

@@ -228,7 +228,7 @@ class AdditiveChi2Sampler(BaseTransformer):
                         inspect.currentframe(), AdditiveChi2Sampler.__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/kernel_approximation/nystroem.py

@@ -276,7 +276,7 @@ class Nystroem(BaseTransformer):
                         inspect.currentframe(), Nystroem.__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/kernel_approximation/polynomial_count_sketch.py

@@ -252,7 +252,7 @@ class PolynomialCountSketch(BaseTransformer):
                         inspect.currentframe(), PolynomialCountSketch.__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/kernel_approximation/rbf_sampler.py

@@ -239,7 +239,7 @@ class RBFSampler(BaseTransformer):
                         inspect.currentframe(), RBFSampler.__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/kernel_approximation/skewed_chi2_sampler.py

@@ -237,7 +237,7 @@ class SkewedChi2Sampler(BaseTransformer):
                         inspect.currentframe(), SkewedChi2Sampler.__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/kernel_ridge/kernel_ridge.py

@@ -273,7 +273,7 @@ class KernelRidge(BaseTransformer):
                         inspect.currentframe(), KernelRidge.__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/lightgbm/lgbm_classifier.py

@@ -262,7 +262,7 @@ class LGBMClassifier(BaseTransformer):
                         inspect.currentframe(), LGBMClassifier.__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/lightgbm/lgbm_regressor.py

@@ -262,7 +262,7 @@ class LGBMRegressor(BaseTransformer):
                         inspect.currentframe(), LGBMRegressor.__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/linear_model/ard_regression.py

@@ -287,7 +287,7 @@ class ARDRegression(BaseTransformer):
                         inspect.currentframe(), ARDRegression.__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/linear_model/bayesian_ridge.py

@@ -298,7 +298,7 @@ class BayesianRidge(BaseTransformer):
                         inspect.currentframe(), BayesianRidge.__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/linear_model/elastic_net.py

@@ -297,7 +297,7 @@ class ElasticNet(BaseTransformer):
                         inspect.currentframe(), ElasticNet.__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/linear_model/elastic_net_cv.py

@@ -333,7 +333,7 @@ class ElasticNetCV(BaseTransformer):
                         inspect.currentframe(), ElasticNetCV.__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/linear_model/gamma_regressor.py

@@ -278,7 +278,7 @@ class GammaRegressor(BaseTransformer):
                         inspect.currentframe(), GammaRegressor.__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/linear_model/huber_regressor.py

@@ -261,7 +261,7 @@ class HuberRegressor(BaseTransformer):
                         inspect.currentframe(), HuberRegressor.__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/linear_model/lars.py

@@ -290,7 +290,7 @@ class Lars(BaseTransformer):
                         inspect.currentframe(), Lars.__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/linear_model/lars_cv.py

@@ -298,7 +298,7 @@ class LarsCV(BaseTransformer):
                         inspect.currentframe(), LarsCV.__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/linear_model/lasso.py

@@ -291,7 +291,7 @@ class Lasso(BaseTransformer):
                         inspect.currentframe(), Lasso.__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/linear_model/lasso_cv.py

@@ -319,7 +319,7 @@ class LassoCV(BaseTransformer):
                         inspect.currentframe(), LassoCV.__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/linear_model/lasso_lars.py

@@ -311,7 +311,7 @@ class LassoLars(BaseTransformer):
                         inspect.currentframe(), LassoLars.__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/linear_model/lasso_lars_cv.py

@@ -312,7 +312,7 @@ class LassoLarsCV(BaseTransformer):
                         inspect.currentframe(), LassoLarsCV.__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/linear_model/lasso_lars_ic.py

@@ -295,7 +295,7 @@ class LassoLarsIC(BaseTransformer):
                         inspect.currentframe(), LassoLarsIC.__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/linear_model/linear_regression.py

@@ -248,7 +248,7 @@ class LinearRegression(BaseTransformer):
                         inspect.currentframe(), LinearRegression.__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/linear_model/logistic_regression.py

@@ -362,7 +362,7 @@ class LogisticRegression(BaseTransformer):
                         inspect.currentframe(), LogisticRegression.__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/linear_model/logistic_regression_cv.py

@@ -383,7 +383,7 @@ class LogisticRegressionCV(BaseTransformer):
                         inspect.currentframe(), LogisticRegressionCV.__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/linear_model/multi_task_elastic_net.py

@@ -281,7 +281,7 @@ class MultiTaskElasticNet(BaseTransformer):
                         inspect.currentframe(), MultiTaskElasticNet.__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/linear_model/multi_task_elastic_net_cv.py

@@ -322,7 +322,7 @@ class MultiTaskElasticNetCV(BaseTransformer):
                         inspect.currentframe(), MultiTaskElasticNetCV.__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/linear_model/multi_task_lasso.py

@@ -273,7 +273,7 @@ class MultiTaskLasso(BaseTransformer):
                         inspect.currentframe(), MultiTaskLasso.__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/linear_model/multi_task_lasso_cv.py

@@ -308,7 +308,7 @@ class MultiTaskLassoCV(BaseTransformer):
                         inspect.currentframe(), MultiTaskLassoCV.__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/linear_model/orthogonal_matching_pursuit.py

@@ -256,7 +256,7 @@ class OrthogonalMatchingPursuit(BaseTransformer):
                         inspect.currentframe(), OrthogonalMatchingPursuit.__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/linear_model/passive_aggressive_classifier.py

@@ -330,7 +330,7 @@ class PassiveAggressiveClassifier(BaseTransformer):
                         inspect.currentframe(), PassiveAggressiveClassifier.__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/linear_model/passive_aggressive_regressor.py

@@ -316,7 +316,7 @@ class PassiveAggressiveRegressor(BaseTransformer):
                         inspect.currentframe(), PassiveAggressiveRegressor.__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/linear_model/perceptron.py

@@ -329,7 +329,7 @@ class Perceptron(BaseTransformer):
                         inspect.currentframe(), Perceptron.__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/linear_model/poisson_regressor.py

@@ -278,7 +278,7 @@ class PoissonRegressor(BaseTransformer):
                         inspect.currentframe(), PoissonRegressor.__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/linear_model/ransac_regressor.py

@@ -334,7 +334,7 @@ class RANSACRegressor(BaseTransformer):
                         inspect.currentframe(), RANSACRegressor.__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/linear_model/ridge.py

@@ -326,7 +326,7 @@ class Ridge(BaseTransformer):
                         inspect.currentframe(), Ridge.__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/linear_model/ridge_classifier.py

@@ -326,7 +326,7 @@ class RidgeClassifier(BaseTransformer):
                         inspect.currentframe(), RidgeClassifier.__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