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

teradataml/analytics/meta_class.py

@@ -192,6 +192,45 @@ def _common_init(self, function_type, **kwargs):
     elif function_type == 'byom':
         from teradataml.analytics.analytic_function_executor import _BYOMFunctionExecutor
         self.obj = _BYOMFunctionExecutor(self.__class__.__name__)._execute_function(**kwargs)
+    elif function_type == 'stored_procedure':
+        from teradataml.analytics.analytic_function_executor import _StoredProcedureExecutor
+        self.obj = _StoredProcedureExecutor(self.__class__.__name__)._execute_function(**kwargs)
     else:
         from teradataml.analytics.analytic_function_executor import _TableOperatorExecutor
-        self.obj = _TableOperatorExecutor(self.__class__.__name__)._execute_function(**kwargs)
+        self.obj = _TableOperatorExecutor(self.__class__.__name__)._execute_function(**kwargs)
+
+def _common_dir(self):
+    """
+    DESCRIPTION:
+        Function returns the attributes and/or names of the methods of the
+        Analytic function.
+
+    PARAMETERS:
+        None
+
+    RETURNS:
+        list
+
+    RAISES:
+        None
+
+    Examples:
+        # Load the data.
+        titanic_data = DataFrame.from_table("titanic")
+        bin_fit_ip = DataFrame.from_table("bin_fit_ip")
+
+        # Run the function.
+        bin_code_1 = BincodeFit(data=titanic_data,
+                                fit_data=bin_fit_ip,
+                                fit_data_order_column = ['minVal', 'maxVal'],
+                                target_columns='age',
+                                minvalue_column='minVal',
+                                maxvalue_column='maxVal',
+                                label_column='label',
+                                method_type='Variable-Width',
+                                label_prefix='label_prefix'
+                                )
+        # Run dir on the function.
+        dir(bin_code_1)
+    """
+    return [attr for attr in super(self.__class__, self).__dir__() if attr != 'obj']

teradataml/analytics/sqle/DecisionTreePredict.py

@@ -223,7 +223,7 @@ class DecisionTreePredict:
         
         # Check to make sure input table types are strings or data frame objects or of valid type.
         self.__awu._validate_input_table_datatype(self.newdata, "newdata", None)
-        self.__awu._validate_input_table_datatype(self.object, "object", DataFrame)
+        self.__awu._validate_input_table_datatype(self.object, "object", reference_function_name=None)
         
         # Check whether the input columns passed to the argument are not empty.
         # Also check whether the input columns passed to the argument valid or not.

teradataml/analytics/sqle/__init__.py

@@ -2,7 +2,7 @@ from teradataml.analytics.sqle.DecisionTreePredict import DecisionTreePredict
 from teradataml.analytics.sqle.NaiveBayesPredict import NaiveBayesPredict
 
 from teradataml.analytics.meta_class import _AnalyticFunction
-from teradataml.analytics.meta_class import _common_init
+from teradataml.analytics.meta_class import _common_init, _common_dir
 from teradataml.analytics.json_parser.utils import _get_associated_parent_classes
 
 _sqle_functions = ['ANOVA',
@@ -11,6 +11,7 @@ _sqle_functions = ['ANOVA',
                   'BincodeFit',
                   'BincodeTransform',
                   'CategoricalSummary',
+                  'CFilter',
                   'ChiSq',
                   'ClassificationEvaluator',
                   'ColumnSummary',
@@ -33,6 +34,8 @@ _sqle_functions = ['ANOVA',
                   'KMeansPredict',
                   'KNN',
                   'MovingAverage',
+                  'NaiveBayes',
+                  'TDNaiveBayesPredict',
                   'NaiveBayesTextClassifierPredict',
                   'NaiveBayesTextClassifierTrainer',
                   'NGramSplitter',
@@ -49,6 +52,7 @@ _sqle_functions = ['ANOVA',
                   'OutlierFilterFit',
                   'OutlierFilterTransform',
                   'Pack',
+                  'Pivoting',
                   'PolynomialFeaturesFit',
                   'PolynomialFeaturesTransform',
                   'QQNorm',
@@ -67,20 +71,24 @@ _sqle_functions = ['ANOVA',
                   'Silhouette',
                   'SimpleImputeFit',
                   'SimpleImputeTransform',
+                  'SMOTE',
                   'StrApply',
                   'StringSimilarity',
                   'SVM',
                   'SVMPredict',
                   'SVMSparsePredict',
+                  'Shap',
                   'TDGLMPredict',
                   'TDDecisionForestPredict',
                   'TargetEncodingFit',
                   'TargetEncodingTransform',
                   'TextParser',
+                  'TFIDF',
                   'Transform',
                   'TrainTestSplit',
                   'UnivariateStatistics',
                   'Unpack',
+                  'Unpivoting',
                   'VectorDistance',
                   'WhichMax',
                   'WhichMin',
@@ -95,4 +103,5 @@ for func in _sqle_functions:
     for assoc_cl in _get_associated_parent_classes(func):
         _c = _c + (assoc_cl, )
     globals()[func] = type("{}".format(func), _c, {"__init__": lambda self, **kwargs: _common_init(self, 'sqle',
-                                                              **kwargs), "__doc__": _AnalyticFunction.__doc__})
+                                                              **kwargs), "__doc__": _AnalyticFunction.__doc__,
+                                                   "__dir__": _common_dir})

teradataml/analytics/table_operator/__init__.py

@@ -1,10 +1,11 @@
 from teradataml.analytics.meta_class import _AnalyticFunction
-from teradataml.analytics.meta_class import _common_init
+from teradataml.analytics.meta_class import _common_init, _common_dir
 
-_nos_functions = ['ReadNOS', 'WriteNOS']
+_nos_functions = ['ReadNOS', 'WriteNOS', 'Image2Matrix']
 
 for func in _nos_functions:
     globals()[func] = type("{}".format(func), (_AnalyticFunction,),
                            {"__init__": lambda self, **kwargs: _common_init(self,
                             'nos', **kwargs),
-                            "__doc__": _AnalyticFunction.__doc__})
+                            "__doc__": _AnalyticFunction.__doc__,
+                            "__dir__": _common_dir})

teradataml/analytics/uaf/__init__.py

@@ -1,5 +1,5 @@
 from teradataml.analytics.meta_class import _AnalyticFunction
-from teradataml.analytics.meta_class import _common_init
+from teradataml.analytics.meta_class import _common_init, _common_dir
 from teradataml.analytics.json_parser.utils import _get_associated_parent_classes
 
 _uaf_functions = ['ACF',
@@ -51,7 +51,17 @@ _uaf_functions = ['ACF',
                   'InputValidator',
                   'MInfo',
                   'SInfo',
-                  'TrackingOp']
+                  'TrackingOp',
+                  'AutoArima',
+                  'ArimaXEstimate',
+                  'DWT',
+                  'DWT2D',
+                  'IDWT',
+                  'IDWT2D',
+                  'IQR',
+                  'Matrix2Image',
+                  'SAX',
+                  'WindowDFFT']
 
 for func in _uaf_functions:
     _c = (_AnalyticFunction,)
@@ -60,4 +70,13 @@ for func in _uaf_functions:
     globals()[func] = type("{}".format(func), _c,
                            {"__init__": lambda self, **kwargs: _common_init(self,
                             'uaf', **kwargs),
+                            "__doc__": _AnalyticFunction.__doc__,
+                            "__dir__": _common_dir})
+
+_stored_procedure = ['CopyArt', 'FilterFactory1d']
+
+for func in _stored_procedure:
+    globals()[func] = type("{}".format(func), (_AnalyticFunction,),
+                           {"__init__": lambda self, **kwargs: _common_init(self,
+                            'stored_procedure', **kwargs),
                             "__doc__": _AnalyticFunction.__doc__})

teradataml/analytics/utils.py

@@ -434,7 +434,29 @@ class FuncSpecialCaseHandler():
                            "NPath": {"mode": self._avoid_quote_for_arg,
                                      "symbols": self._avoid_quote_for_arg,
                                      "result": self._avoid_quote_for_arg,
-                                     "filter": self._avoid_quote_for_arg}}
+                                     "filter": self._avoid_quote_for_arg},
+                           "Pivoting": {"combined_column_sizes": self._handle_multiple_datatype},
+                           "FilterFactory1d": {"database_name": self._single_quote_arg,
+                                               "table_name": self._single_quote_arg,
+                                               "filter_type": self._single_quote_arg,
+                                               "window_type": self._single_quote_arg,
+                                               "filter_description": self._single_quote_arg},
+                           "CopyArt":{"database_name": self._single_quote_arg,
+                                      "table_name": self._single_quote_arg,
+                                      "map_name": self._single_quote_arg,
+                                      "permanent_table": self._single_quote_arg},
+                           "DWT": {"wavelet": self._single_quote_arg},
+                           "IDWT": {"part": self._single_quote_arg,
+                                    "wavelet": self._single_quote_arg,
+                                    "mode": self._single_quote_arg},
+                           "DWT2D": {"wavelet": self._single_quote_arg,
+                                     "mode": self._single_quote_arg},
+                           "IDWT2D": {"wavelet": self._single_quote_arg,
+                                      "mode": self._single_quote_arg},
+                           "Matrix2Image": {"type": self._single_quote_arg,
+                                            "colormap": self._single_quote_arg
+                                           }
+                           }
 
     # Setter method for argument.
     def set_arg_name(self, argument):
@@ -691,4 +713,47 @@ class FuncSpecialCaseHandler():
                     if sep in value and "-" != value[0]:
                         arg_value[num] = "[{}]".format(arg_value[num])
                         break
+        return arg_value
+    
+    def _handle_multiple_datatype(self, arg_value, *args, **kwargs):
+
+        """
+        DESCRIPTION:
+            Function to handle multiple data types in "arg_value".
+                * Function returns the "arg_value" in sqle expected format.
+
+        PARAMETERS:
+            arg_value:
+                Required Argument.
+                Specifies the arg_value to handle multiple data types.
+                Types: int or str or list of str(s)
+
+        RETURNS:
+            arg_value converted into sqle expected format.
+
+        RAISES:
+            None
+
+        EXAMPLES:
+            # Scenario 1: convert list of string into sqle expected format.
+                self._handle_multiple_datatype(arg_value)
+                # returns "'col1:1234','col2:1144','col3:214'"
+
+            # Scenario 2: convert string into sqle expected format.
+                self._handle_multiple_datatype(arg_value)
+                # returns "'col1:1234'"
+        """
+        # Datatype of arg_value already validated inside function.
+
+        # Convert the pass value into sqle expected format.
+        # If argument value is a list, convert into sqle expected string format.
+        # If argument value is a string, add single quote.
+        # If argument value is a integer, return as it is.
+        if isinstance(arg_value, list):
+            if all(isinstance(i, str) for i in arg_value):
+                arg_value = "'" + "','".join(arg_value) + "'"
+        elif isinstance(arg_value, str):
+            arg_value = [arg_value]
+            arg_value = "'" + "','".join(arg_value) + "'"
+
         return arg_value

teradataml/analytics/valib.py

@@ -13,7 +13,7 @@ import time
 import uuid
 from math import floor
 
-from teradatasqlalchemy.telemetry.queryband import collect_queryband
+from teradataml.telemetry_utils.queryband import collect_queryband
 from teradataml.common import messages
 from teradataml.common.constants import TeradataConstants, ValibConstants as VC
 from teradataml.common.exceptions import TeradataMlException