teradataml 20.0.0.3__py3-none-any.whl → 20.0.0.5__py3-none-any.whl

teradataml/automl/model_training.py

@@ -26,10 +26,10 @@ from teradataml.context import context as tdmlctx
 from teradataml.dataframe.copy_to import copy_to_sql
 from teradataml.dataframe.dataframe import DataFrame
 from teradataml import execute_sql, get_connection
-from teradataml import SVM, GLM, DecisionForest, XGBoost, GridSearch, KNN, RandomSearch
+from teradataml import configure, SVM, GLM, DecisionForest, XGBoost, GridSearch, KNN, RandomSearch
 from teradataml.utils.validators import _Validators
 from teradataml.common.utils import UtilFuncs
-
+from teradataml.common.constants import TeradataConstants
 
 class _ModelTraining:
     
@@ -114,6 +114,12 @@ class _ModelTraining:
                         session.
                         Default Value: False
                         Types: bool
+                    
+                    seed:
+                        Optional Argument.
+                        Specifies the random seed for reproducibility.
+                        Default Value: 42
+                        Types: int
         """
         self.data = data
         self.target_column = target_column
@@ -126,6 +132,7 @@ class _ModelTraining:
         self.startify_col = None
         self.persist = kwargs.get("persist", False)
         self.volatile = kwargs.get("volatile", False)
+        self.seed = kwargs.get("seed", 42)
 
     def model_training(self, 
                        auto=True,
@@ -499,7 +506,7 @@ class _ModelTraining:
                 'max_depth': tuple(max_depth),
                 'min_node_size': tuple(min_node_size),
                 'iter_num': tuple(iter_num),
-                'seed':42
+                'seed':self.seed
                 }
         # Hyperparameters for Decision Forest model
         df_params = {
@@ -510,7 +517,7 @@ class _ModelTraining:
                 'max_depth': tuple(max_depth),
                 'min_node_size': tuple(min_node_size),
                 'num_trees': tuple(num_trees),
-                'seed':42
+                'seed':self.seed
         }
         
         # Updating model type in case of classification
@@ -768,7 +775,7 @@ class _ModelTraining:
 
         # Defining training data
         data_types = ['lasso', 'rfe', 'pca']
-        trainng_datas = tuple(DataFrame(self.table_name_mapping[f'{data_type}_train']) for data_type in data_types)
+        trainng_datas = tuple(DataFrame(self.data_mapping[f'{data_type}_train']) for data_type in data_types)
 
         if self.task_type == "Classification":
             response_values = trainng_datas[0].get(self.target_column).drop_duplicate().get_values().flatten().tolist()
@@ -874,16 +881,30 @@ class _ModelTraining:
             verbose = 0
             
         # Hyperparameter tunning
+        # Parallel run opens multiple connections for parallel execution, 
+        # but volatile tables are not accessible across different sessions. 
+        # Therefore, execution is performed sequentially by setting run_parallel=False.
+
+        run_parallel = configure.temp_object_type != TeradataConstants.TERADATA_VOLATILE_TABLE
+
+        common_params = {
+            "data": train_data,
+            "evaluation_metric": self.stopping_metric,
+            "early_stop": self.stopping_tolerance,
+            "run_parallel": run_parallel,
+            "sample_seed": self.seed,
+            "sample_id_column": "id",
+            "discard_invalid_column_params": True,
+            "stratify_column": self.startify_col,
+            "verbose": verbose,
+            "max_time": self.max_runtime_secs,
+            "suppress_refer_msg": True
+        }
+        
         if model_param['name'] == 'knn':
-            _obj.fit(data=train_data, evaluation_metric=self.stopping_metric,
-                    early_stop=self.stopping_tolerance, run_parallel=True, 
-                    sample_seed=42, sample_id_column='id', discard_invalid_column_params=True,
-                    stratify_column=self.startify_col,verbose=verbose, max_time=self.max_runtime_secs)
+            _obj.fit(**common_params)
         else:
-            _obj.fit(data=train_data, evaluation_metric=self.stopping_metric,
-                    early_stop=self.stopping_tolerance, **eval_params, 
-                    run_parallel=True, discard_invalid_column_params=True, sample_seed=42, 
-                    sample_id_column='id',stratify_column=self.startify_col, verbose=verbose, max_time=self.max_runtime_secs)
+            _obj.fit(**common_params, **eval_params)
 
         # Getting all passed models
         model_info = _obj.model_stats.merge(_obj.models[_obj.models['STATUS']=='PASS'][['MODEL_ID', 'DATA_ID', 'PARAMETERS']],
@@ -992,4 +1013,4 @@ class _ModelTraining:
             elif ml_name == 'glm':
                 eval_params['family'] = 'GAUSSIAN'
                 
-        return eval_params
+        return eval_params

teradataml/clients/auth_client.py

@@ -1,6 +1,6 @@
 """
 Unpublished work.
-Copyright (c) 2024 by Teradata Corporation. All rights reserved.
+Copyright (c) 2025 by Teradata Corporation. All rights reserved.
 TERADATA CORPORATION CONFIDENTIAL AND TRADE SECRET
 Primary Owner: gouri.patwardhan@teradata.com
 Secondary Owner: Pradeep.Garre@teradata.com
@@ -67,7 +67,7 @@ class _AuthWorkflow:
             dict
         """
         # Extract the pem file name without extension.
-        kid = pathlib.Path(self.state.get('pem_file')).stem
+        kid = pathlib.Path(self.state.get('pem_file')).stem if not self.state.get('kid') else self.state['kid']
         header = {
             "alg": "RS256",
             "kid": kid,

teradataml/common/__init__.py

@@ -1,2 +1 @@
-from teradataml.common.formula import as_categorical
-from teradataml.common.constants import Action, Permission
+from teradataml.common.formula import as_categorical

teradataml/common/constants.py

@@ -14,14 +14,13 @@ A class for holding all constants
 import re
 import sqlalchemy
 from enum import Enum
-from teradataml.options.configure import configure
 from teradatasqlalchemy.types import (INTEGER, SMALLINT, BIGINT, BYTEINT, DECIMAL, FLOAT, NUMBER, VARCHAR)
 from teradatasqlalchemy.types import (DATE, TIME, TIMESTAMP)
 from teradatasqlalchemy.types import (BYTE, VARBYTE, BLOB)
 from teradatasqlalchemy import (CHAR, CLOB)
 from teradatasqlalchemy import (PERIOD_DATE, PERIOD_TIME, PERIOD_TIMESTAMP)
 from teradatasqlalchemy import (INTERVAL_YEAR, INTERVAL_YEAR_TO_MONTH, INTERVAL_MONTH,
-                                INTERVAL_DAY,INTERVAL_DAY_TO_HOUR, INTERVAL_DAY_TO_MINUTE,
+                                INTERVAL_DAY, INTERVAL_DAY_TO_HOUR, INTERVAL_DAY_TO_MINUTE,
                                 INTERVAL_DAY_TO_SECOND, INTERVAL_HOUR,
                                 INTERVAL_HOUR_TO_MINUTE, INTERVAL_HOUR_TO_SECOND,
                                 INTERVAL_MINUTE, INTERVAL_MINUTE_TO_SECOND,
@@ -62,6 +61,8 @@ class SQLConstants(Enum):
     SQL_DELETE_SPECIFIC_ROW = 30
     SQL_EXEC_STORED_PROCEDURE = 31
     SQL_SELECT_COLUMNNAMES_WITH_WHERE = 32
+    SQL_HELP_DATABASE = 33
+    SQL_HELP_DATALAKE = 34
     CONSTRAINT = ["check_constraint", "primary_key_constraint",
                   "foreign_key_constraint", "unique_key_constraint"]
 
@@ -81,7 +82,7 @@ class TeradataConstants(Enum):
     # Order of operators
     # shouldn't be changed. This is the order in which join condition is tested - first, operators
     # with two characters and then the operators with single character.
-    SUPPORTED_ENGINES = {"ENGINE_SQL" : {"name" : "sqle", "file" : "sqlengine_alias_definitions"}}
+    SUPPORTED_ENGINES = {"ENGINE_SQL": {"name": "sqle", "file": "sqlengine_alias_definitions"}}
     SUPPORTED_VANTAGE_VERSIONS = {"vantage1.0": "v1.0", "vantage1.1": "v1.1",
                                   "vantage1.3": "v1.3", "vantage2.0": "v1.1"}
     RANGE_SEPARATORS = [":"]
@@ -89,7 +90,7 @@ class TeradataConstants(Enum):
 
 class AEDConstants(Enum):
     AED_NODE_NOT_EXECUTED = 0
-    AED_NODE_EXECUTED     = 1
+    AED_NODE_EXECUTED = 1
     AED_DB_OBJECT_NAME_BUFFER_SIZE = 128
     AED_NODE_TYPE_BUFFER_SIZE = 32
     AED_ASSIGN_DROP_EXISITING_COLUMNS = "Y"
@@ -147,7 +148,7 @@ class TeradataTableKindConstants(Enum):
     TABLE = "table"
     VIEW = "view"
     TEMP = "temp"
-    ALL  = "all"
+    ALL = "all"
     ML_PATTERN = "ml_%"
     VOLATILE_TABLE_NAME = 'Table Name'
     REGULAR_TABLE_NAME = 'TableName'
@@ -219,7 +220,7 @@ class ModelCatalogingConstants(Enum):
 
 
 class CopyToConstants(Enum):
-    DBAPI_BATCHSIZE                      = 16383
+    DBAPI_BATCHSIZE = 16383
 
 
 class PTITableConstants(Enum):
@@ -376,34 +377,34 @@ class GeospatialConstants(Enum):
         "buffer": lambda x: "ST_Buffer",
         "contains": lambda x: "ST_Contains",
         "crosses": lambda x: "ST_Crosses",
-        "difference": lambda x: "ST_Difference", # M
+        "difference": lambda x: "ST_Difference",  # M
         "disjoint": lambda x: "ST_Disjoint",
-        "distance": lambda x: "ST_Distance", # M
-        "distance_3D": lambda x: "ST_3DDistance", # M
+        "distance": lambda x: "ST_Distance",  # M
+        "distance_3D": lambda x: "ST_3DDistance",  # M
         "geom_equals": lambda x: "ST_Equals",
         "intersection": lambda x: "ST_Intersection",
-        #"intersect": lambda x: "ST_Intersect", # M
+        # "intersect": lambda x: "ST_Intersect", # M
         "make_2D": lambda x: "Make_2D",
         "overlaps": lambda x: "ST_Overlaps",
         "relates": lambda x: "ST_Relate",
         "simplify": lambda x: "SimplifyPreserveTopology",
-        "sym_difference": lambda x: "ST_SymDifference", # M
+        "sym_difference": lambda x: "ST_SymDifference",  # M
         "touches": lambda x: "ST_Touches",
         "transform": lambda x: "ST_Transform",
         "union": lambda x: "ST_Union",
         "within": lambda x: "ST_Within",
-        "wkb_geom_to_sql": lambda x: "ST_WKBToSQL", # M
-        "wkt_geom_to_sql": lambda x: "ST_WKTToSQL", # M
+        "wkb_geom_to_sql": lambda x: "ST_WKBToSQL",  # M
+        "wkt_geom_to_sql": lambda x: "ST_WKTToSQL",  # M
         "set_srid": lambda x: "ST_SRID",
 
         ## *** Geometry Type ST_Point Methods *** ##
         "set_x": lambda x: "ST_X",
         "set_y": lambda x: "ST_Y",
         "set_z": lambda x: "ST_Z",
-        "spherical_buffer": lambda x: "ST_SphericalBufferMBR", # M
-        "spherical_distance": lambda x: "ST_SphericalDistance", # M
-        "spheroidal_buffer": lambda x: "ST_SpheroidalBufferMBR", # M
-        "spheroidal_distance": lambda x: "ST_SpheroidalDistance", # M
+        "spherical_buffer": lambda x: "ST_SphericalBufferMBR",  # M
+        "spherical_distance": lambda x: "ST_SphericalDistance",  # M
+        "spheroidal_buffer": lambda x: "ST_SpheroidalBufferMBR",  # M
+        "spheroidal_distance": lambda x: "ST_SpheroidalDistance",  # M
 
         ## *** Geometry Type ST_LineString Methods *** ##
         "line_interpolate_point": lambda x: "ST_Line_Interpolate_Point",
@@ -485,11 +486,18 @@ class TableOperatorConstants(Enum):
 
     # Check if Python interpretor and add-ons are installed or not.
     # Location of In-DB packages is indicated by configure.indb_install_location.
+    # Check for both python and pip versions.
     CHECK_PYTHON_INSTALLED = """SELECT distinct * FROM SCRIPT(
                                 ON (select 1) PARTITION BY ANY
-                                SCRIPT_COMMAND('{}/bin/pip3 --version')
-                                returns('package VARCHAR(256)'))
+                                SCRIPT_COMMAND('echo $({0}/bin/pip3 --version) -- $({0}/bin/python3 --version)')
+                                returns('pip VARCHAR(256)'))
                              """
+    # Check which version of rpms are installed.
+    INDB_PYTHON_PATH = """SEL DISTINCT os_ver
+            FROM SCRIPT(
+                SCRIPT_COMMAND('grep CPE_NAME /etc/os-release')
+                RETURNS('os_ver VARCHAR(100)')
+            );"""
 
     # Script Query to get Python packages and corresponding versions.
     # Location of In-DB packages is indicated by configure.indb_install_location.
@@ -501,9 +509,18 @@ class TableOperatorConstants(Enum):
                                                     "delimiter(' ') " \
                                                     "returns('package VARCHAR({2}), " \
                                                     "version VARCHAR({2})'))"
-    
+
     SCRIPT_LIST_FILES_QUERY = "SELECT DISTINCT * FROM SCRIPT (SCRIPT_COMMAND " \
-                       "('ls ./{}') RETURNS ('Files VARCHAR({})'))"
+                              "('ls ./{}') RETURNS ('Files VARCHAR({})'))"
+
+
+    # OpenBlas by default is multi-threaded, needs to be set to single-threaded.
+    OPENBLAS_NUM_THREADS = "OPENBLAS_NUM_THREADS=1"
+
+    # Query to create a DataFrame with a range of numbers.
+    RANGE_QUERY = "WITH RECURSIVE NumberSeries (id) AS (SELECT id AS id from {0} "\
+                  "UNION ALL SELECT id {3} {1} FROM NumberSeries WHERE id {3} {1} {4} {2}) "\
+                  "SELECT id FROM NumberSeries;"
 
 class ValibConstants(Enum):
     # A dictionary that maps teradataml name of the exposed VALIB function name
@@ -802,7 +819,8 @@ class ValibConstants(Enum):
             "subdivision_method": "subdivisionmethod",
             "subdivision_threshold": "subdivisionthreshold",
             "filter": "where",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "DATAEXPLORER": {
@@ -819,7 +837,8 @@ class ValibConstants(Enum):
             "stats_options": "statsoptions",
             "distinct": "uniques",
             "filter": "where",
-            "gen_sql": "gensql"
+            "gen_sql": "gensql",
+            "charset": "charset"
         },
 
         "FREQUENCY": {
@@ -833,7 +852,8 @@ class ValibConstants(Enum):
             "style": "style",
             "top_n": "topvalues",
             "filter": "where",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "HISTOGRAM": {
@@ -848,7 +868,8 @@ class ValibConstants(Enum):
             "stats_columns": "statisticscolumns",
             "hist_style": "style",
             "filter": "where",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "STATISTICS": {
@@ -859,7 +880,8 @@ class ValibConstants(Enum):
             "statistical_method": "statisticalmethod",
             "stats_options": "statsoptions",
             "filter": "where",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "TEXTFIELDANALYZER": {
@@ -867,7 +889,8 @@ class ValibConstants(Enum):
             "exclude_columns": "columnstoexclude",
             "analyze_numerics": "extendednumericanalysis",
             "analyze_unicode": "extendedunicodeanalysis",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "VALUES": {
@@ -876,7 +899,8 @@ class ValibConstants(Enum):
             "group_columns": "groupby",
             "distinct": "uniques",
             "filter": "where",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "ASSOCIATION": {
@@ -901,7 +925,8 @@ class ValibConstants(Enum):
             "filter": "where",
             "no_support_results": "dropsupporttables",
             "support_result_prefix": "resulttableprefix",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "KMEANS": {
@@ -911,7 +936,8 @@ class ValibConstants(Enum):
             "continuation": "continuation",
             "max_iter": "iterations",
             "operator_database": "operatordatabase",
-            "threshold": "threshold"
+            "threshold": "threshold",
+            "charset": "charset"
         },
 
         "KMEANSSCORE": {
@@ -919,7 +945,8 @@ class ValibConstants(Enum):
             "cluster_column": "clustername",
             "fallback": "fallback",
             "operator_database": "operatordatabase",
-            "accumulate": "retain"
+            "accumulate": "retain",
+            "charset": "charset"
         },
 
         "DECISIONTREE": {
@@ -931,7 +958,8 @@ class ValibConstants(Enum):
             "max_depth": "max_depth",
             "num_splits": "min_records",
             "operator_database": "operatordatabase",
-            "pruning": "pruning"
+            "pruning": "pruning",
+            "charset": "charset"
         },
 
         "DECISIONTREESCORE": {
@@ -941,7 +969,8 @@ class ValibConstants(Enum):
             "profile": "profiletables",
             "accumulate": "retain",
             "targeted_value": "targetedvalue",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "MATRIX": {
@@ -951,7 +980,8 @@ class ValibConstants(Enum):
             "matrix_output": "matrixoutput",
             "type": "matrixtype",
             "handle_nulls": "nullhandling",
-            "filter": "where"
+            "filter": "where",
+            "charset": "charset"
         },
 
         "LINEAR": {
@@ -973,7 +1003,8 @@ class ValibConstants(Enum):
             "stepwise": "stepwise",
             "use_fstat": "usefstat",
             "use_pvalue": "usepvalue",
-            "variance_prop_threshold": "varianceproportionthreshold"
+            "variance_prop_threshold": "varianceproportionthreshold",
+            "charset": "charset"
         },
 
         "LINEARSCORE": {
@@ -981,7 +1012,8 @@ class ValibConstants(Enum):
             "response_column": "predicted",
             "residual_column": "residual",
             "accumulate": "retain",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "LOGISTIC": {
@@ -1011,7 +1043,8 @@ class ValibConstants(Enum):
             "end_threshold": "thresholdend",
             "increment_threshold": "thresholdincrement",
             "threshold_output": "thresholdtable",
-            "variance_prop_threshold": "varianceproportionthreshold"
+            "variance_prop_threshold": "varianceproportionthreshold",
+            "charset": "charset"
         },
 
         "LOGISTICSCORE": {
@@ -1023,7 +1056,8 @@ class ValibConstants(Enum):
             "start_threshold": "thresholdbegin",
             "end_threshold": "thresholdend",
             "increment_threshold": "thresholdincrement",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
 
             # The following 3 arguments three should not be present for LogRegPredict function
             # where as when the function is LogRegEvaluator, at least one of these should be
@@ -1051,13 +1085,15 @@ class ValibConstants(Enum):
             "rotation_type": "rotationtype",
             "load_threshold": "thresholdloading",
             "percent_threshold": "thresholdpercent",
-            "variance_prop_threshold": "varianceproportionthreshold"
+            "variance_prop_threshold": "varianceproportionthreshold",
+            "charset": "charset"
         },
 
         "FACTORSCORE": {
             "index_columns": "index",
             "accumulate": "retain",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "PARAMETRICTEST": {
@@ -1076,7 +1112,8 @@ class ValibConstants(Enum):
             "style": "teststyle",
             "probability_threshold": "thresholdprobability",
             "with_indicator": "withindicator",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "BINOMIALTEST": {
@@ -1091,7 +1128,8 @@ class ValibConstants(Enum):
             "stats_database": "statsdatabase",
             "style": "teststyle",
             "probability_threshold": "thresholdprobability",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "KSTEST": {
@@ -1103,7 +1141,8 @@ class ValibConstants(Enum):
             "stats_database": "statsdatabase",
             "style": "teststyle",
             "probability_threshold": "thresholdprobability",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "CHISQUARETEST": {
@@ -1117,7 +1156,8 @@ class ValibConstants(Enum):
             "stats_database": "statsdatabase",
             "style": "teststyle",
             "probability_threshold": "thresholdprobability",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "RANKTEST": {
@@ -1136,7 +1176,8 @@ class ValibConstants(Enum):
             "style": "teststyle",
             "probability_threshold": "thresholdprobability",
             "treatment_column": "treatmentcolumn",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "VARTRAN": {
@@ -1147,13 +1188,15 @@ class ValibConstants(Enum):
             "allow_duplicates": "multiset",
             "nopi": "noindex",
             "filter": "whereclause",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         },
 
         "REPORT": {
             "analysis_type": "analysistype",
             "filter": "where",
-            "gen_sql_only": "gensqlonly"
+            "gen_sql_only": "gensqlonly",
+            "charset": "charset"
         }
     }
 
@@ -1321,7 +1364,7 @@ class SQLFunctionConstants(Enum):
         "regexp_replace": "REGEXP_REPLACE",
         "regexp_similar": "REGEXP_SIMILAR",
         "regexp_substr": "REGEXP_SUBSTR",
-        
+
         # DateTime Functions
         'week_begin': 'td_week_begin',
         'week_start': 'td_week_begin',
@@ -1392,6 +1435,7 @@ class TeradataReservedKeywords(Enum):
                                "TYPE"
                                ]
 
+
 class TeradataAnalyticFunctionTypes(Enum):
     SQLE = "FASTPATH"
     UAF = "UAF"
@@ -1401,15 +1445,15 @@ class TeradataAnalyticFunctionTypes(Enum):
 
 
 class TeradataAnalyticFunctionInfo(Enum):
-
-    FASTPATH = {"func_type": "sqle", "lowest_version": "16.20", "display_function_type_name" :"SQLE"}
+    FASTPATH = {"func_type": "sqle", "lowest_version": "16.20", "display_function_type_name": "SQLE"}
     UAF = {"func_type": "uaf", "lowest_version": "17.20", "display_function_type_name": "UAF",
-           "metadata_class" : "_AnlyFuncMetadataUAF"}
+           "metadata_class": "_AnlyFuncMetadataUAF"}
     TABLE_OPERATOR = {"func_type": "tableoperator", "lowest_version": "17.00 ",
-                      "display_function_type_name" :"TABLE OPERATOR"}
+                      "display_function_type_name": "TABLE OPERATOR"}
     BYOM = {"func_type": "byom", "lowest_version": None, "display_function_type_name": "BYOM"}
     STORED_PROCEDURE = {"func_type": "storedprocedure", "lowest_version": "17.20", "display_function_type_name": "UAF",
-           "metadata_class" : "_AnlyFuncMetadataUAF"}
+                        "metadata_class": "_AnlyFuncMetadataUAF"}
+
 
 class TeradataUAFSpecificArgs(Enum):
     INPUT_MODE = "input_mode"
@@ -1417,9 +1461,11 @@ class TeradataUAFSpecificArgs(Enum):
     OUTPUT_FMT_INDEX = "output_fmt_index"
     OUTPUT_FMT_INDEX_STYLE = "output_fmt_index_style"
 
+
 class Query(Enum):
     VANTAGE_VERSION = "SELECT InfoData FROM DBC.DBCInfoV where InfoKey = 'VERSION'"
 
+
 class DriverEscapeFunctions(Enum):
     # Holds variables for the teradatasql driver escape functions to be used
     NATIVE_SQL = "{fn teradata_nativesql}"
@@ -1448,6 +1494,7 @@ class HTTPRequest(Enum):
     POST = "post"
     PUT = "put"
     DELETE = "delete"
+    PATCH = "patch"
 
 
 class AsyncStatusColumns(Enum):
@@ -1459,6 +1506,13 @@ class AsyncStatusColumns(Enum):
     ADDITIONAL_DETAILS = "Additional Details"
 
 
+class AsyncOpStatus(Enum):
+    # Holds valid status for asynchronous operatiosns in UES.
+    FILE_INSTALLED = "File Installed"
+    ERRED = "Errored"
+    FINISHED = "Finished"
+
+
 class CloudProvider(Enum):
     # Holds variable names for Cloud Providers.
     AWS = "AWS"
@@ -1468,7 +1522,8 @@ class CloudProvider(Enum):
     X_MS_VERSION = "2019-12-12"
     X_MS_BLOB_TYPE = "BlockBlob"
 
-class SessionParamsSQL:
+
+class SessionParamsSQL(Enum):
     # Holds the SQL Statements for Session params.
     TIMEZONE = "SET TIME ZONE {}"
     ACCOUNT = "SET SESSION ACCOUNT = '{}' FOR {}"
@@ -1488,7 +1543,8 @@ class SessionParamsSQL:
     QUERY_BAND = "SET QUERY_BAND = {} FOR {}"
     UDFSEARCHPATH = "SET SESSION UDFSEARCHPATH = {} FOR FUNCTION = {}"
 
-class SessionParamsPythonNames:
+
+class SessionParamsPythonNames(Enum):
     # Holds the SQL Statements for Session params.
     TIMEZONE = "Session Time Zone"
     ACCOUNT = "Account Name"
@@ -1498,13 +1554,16 @@ class SessionParamsPythonNames:
     DATEFORM = 'Current DateForm'
 
 
-class Action(Enum):
-    # Holds variable names for the type of grant to be provided.
-    GRANT = "GRANT"
-    REVOKE = "REVOKE"
+class AutoMLConstants(Enum):
+    # List stores feature selection methods
+    FEATURE_SELECTION_MTDS = ["lasso", "rfe", "pca"]
 
-class Permission(Enum):
-    # Holds variable names for the type of permission to be provided.
-    READ = "READ"
-    WRITE = "WRITE"
 
+class AuthMechs(Enum):
+    """
+    Enum to hold permitted values for authentication mechanism.
+    """
+    OAUTH = "OAuth"
+    JWT = "JWT"
+    PAT = "PAT"
+    BASIC = "BASIC"