teradataml 20.0.0.2__py3-none-any.whl → 20.0.0.4__py3-none-any.whl

teradataml/context/context.py

@@ -13,32 +13,35 @@ A teradataml context functions provide interface to Teradata Vantage. Provides f
 context which can be used by other analytical functions to get the Teradata Vantage connection.
 
 """
+import atexit
 import ipaddress
+import os
+import socket
+import sys
+import threading
+import urllib.parse
+import warnings
 from pathlib import Path
+from dotenv import dotenv_values
+
 from sqlalchemy import create_engine
-from teradataml.telemetry_utils.queryband import collect_queryband
+from sqlalchemy.engine.base import Engine
+from sqlalchemy.engine.url import URL
+
+from teradataml.common.constants import Query, SQLConstants, TeradataConstants
 from teradataml.common.deprecations import argument_deprecation
 from teradataml.common.exceptions import TeradataMlException
-from teradataml.common.warnings import TeradataMlRuntimeWarning
-from teradataml.common.messages import Messages
+from teradataml.common.garbagecollector import GarbageCollector
 from teradataml.common.messagecodes import MessageCodes
+from teradataml.common.messages import Messages
 from teradataml.common.sqlbundle import SQLBundle
-from teradataml.common.constants import SQLConstants
-from teradataml.common.garbagecollector import GarbageCollector
+from teradataml.common.warnings import TeradataMlRuntimeWarning
 from teradataml.context.aed_context import AEDContext
-from teradataml.common.constants import TeradataConstants, Query
 from teradataml.options.configure import configure
+from teradataml.telemetry_utils.queryband import collect_queryband
+from teradataml.utils.internal_buffer import _InternalBuffer
 from teradataml.utils.utils import execute_sql
 from teradataml.utils.validators import _Validators
-from teradataml.utils.internal_buffer import _InternalBuffer
-from sqlalchemy.engine.base import Engine
-from sqlalchemy.engine.url import URL
-import os
-import warnings
-import atexit
-import socket
-import threading
-import urllib.parse
 
 # Store a global Teradata Vantage Connection.
 # Right now user can only provide a single Vantage connection at any point of time.
@@ -47,6 +50,8 @@ td_sqlalchemy_engine = None
 temporary_database_name = None
 user_specified_connection = False
 python_packages_installed = False
+python_version_vantage = None
+python_version_local = None
 td_user = None
 
 function_alias_mappings = {}
@@ -201,9 +206,16 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
     """
     DESCRIPTION:
         Creates a connection to the Teradata Vantage using the teradatasql + teradatasqlalchemy DBAPI and dialect
-        combination. Users can pass all required parameters (host, username, password) for establishing a connection to
-        Vantage, or pass a sqlalchemy engine to the tdsqlengine parameter to override the default DBAPI and dialect
         combination.
+        Users can create a connection by passing the connection parameters using any of the following methods:
+            1. Pass all required parameters (host, username, password) directly to the function, OR
+            2. Set the connection parameters in a configuration file (.cfg or .env) and
+               pass the configuration file OR
+            3. Set the connection parameters in environment variables and create_context() reads from
+               environment variables.
+
+        Alternatively, users can pass a SQLAlchemy engine to the `tdsqlengine` parameter to override the default DBAPI
+        and dialect combination.
 
         Note:
             1. teradataml requires that the user has certain permissions on the user's default database or the initial
@@ -241,6 +253,17 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
                |                                                      | default database.                           |
                +------------------------------------------------------+---------------------------------------------+
 
+            3. The function prioritizes parameters in the following order:
+                1. Explicitly passed arguments (host, username, password).
+                2. Environment variables (TD_HOST, TD_USERNAME, TD_PASSWORD, etc.).
+                   Note:
+                       * The environment variables should start with 'TD_' and all must be in upper case.
+                         Example:
+                                os.environ['TD_HOST'] = 'tdhost'
+                                os.environ['TD_USERNAME'] = 'tduser'
+                                os.environ['TD_PASSWORD'] = 'tdpassword'
+                3. A configuration file if provided (such as a .env file).
+
     PARAMETERS:
         host:
             Optional Argument.
@@ -331,6 +354,25 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
                   integer or boolean value, instead of quoted integer or quoted boolean as suggested in the
                   documentation. Please check the examples for usage.
 
+            config_file:
+                Optional Argument.
+                Specifies the name of the configuration file to read the connection parameters.
+                Notes:
+                    * If user do not specify full path of file,then file look up is done at current working directory.
+                    * The content of the file must be in '.env' format.
+                    * Use parameters of create_context() as key in the configuration file.
+                      Example:
+                          host=tdhost
+                          username=tduser
+                          password=tdpassword
+                          temp_database_name=tdtemp_database_name
+                          logmech=tdlogmech
+                          logdata=tdlogdata
+                          database=tddatabase
+                    * For more information please refer examples section.
+                Default Value : td_properties.cfg
+                Types: str
+
     RETURNS:
         A Teradata sqlalchemy engine object.
 
@@ -400,13 +442,56 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
         >>> td_context = create_context(host = 'tdhost', username='tduser', password = 'tdpassword', tmode = 'tera',
                                         log = 8, lob_support = False)
 
+        # Example 14: Create context from config file with name 'td_properties.cfg'
+        #             available under current working directory.
+        #             td_properties.cfg content:
+        #                 host=tdhost
+        #                 username=tduser
+        #                 password=tdpassword
+        #                 temp_database_name=tdtemp_database_name
+        #                 logmech=tdlogmech
+        #                 logdata=tdlogdata
+        #                 database=tddatabase
+        >>> td_context = create_context()
+
+        # Example 15: Create context using the file specified in user's home directory
+        #             with name user_td_properties.cfg.
+        #             user_td_properties.cfg content:
+        #                 host=tdhost
+        #                 username=tduser
+        #                 password=tdpassword
+        #                 temp_database_name=tdtemp_database_name
+        #                 logmech=tdlogmech
+        #                 logdata=tdlogdata
+        #                 database=tddatabase
+        >>> td_context = create_context(config_file="user_td_properties.cfg")
+
+        # Example 16: Create context using environment variables.
+        #             Set these using os.environ and then run the example:
+        #                  os.environ['TD_HOST'] = 'tdhost'
+        #                  os.environ['TD_USERNAME'] = 'tduser'
+        #                  os.environ['TD_PASSWORD'] = 'tdpassword'
+        #                  os.environ['TD_TEMP_DATABASE_NAME'] = 'tdtemp_database_name'
+        #                  os.environ['TD_LOGMECH'] = 'tdlogmech'
+        #                  os.environ['TD_LOGDATA'] = 'tdlogdata'
+        #                  os.environ['TD_DATABASE'] = 'tddatabase'
+        >>> td_context = create_context()
     """
     global td_connection
     global td_sqlalchemy_engine
     global temporary_database_name
     global user_specified_connection
     global python_packages_installed
+    global python_version_vantage
+    global python_version_local
     global td_user
+
+    # Check if the user has provided the connection parameters or tdsqlengine.
+    # If not, check if the user has provided the connection parameters in the environment variables.
+    # If not, check if the user has provided the connection parameters in the config file.
+    if not (host or tdsqlengine) and host!="":
+        return _load_context_from_env_config(kwargs.pop('config_file', 'td_properties.cfg'))
+
     awu_matrix = []
     awu_matrix.append(["host", host, True, (str), True])
     awu_matrix.append(["username", username, True, (str), True])
@@ -497,6 +582,8 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
     _load_function_aliases()
 
     python_packages_installed = False
+    python_version_vantage = None
+    python_version_local = sys.version.split(" ")[0].strip()
 
     # Assign the tempdatabase name to global
     if temp_database_name is None:
@@ -527,6 +614,34 @@ def create_context(host = None, username = None, password = None, tdsqlengine =
     # Return the connection by default
     return td_sqlalchemy_engine
 
+def _load_context_from_env_config(config_file):
+    """
+    DESCRIPTION:
+        Reads the connection parameters from the configuration file or environment variables.
+
+    PARAMETERS:
+        config_file:
+            Required Argument.
+            Specifies the name of the configuration file to read the connection parameters.
+            Types: str
+
+    RETURNS:
+        A Teradata sqlalchemy engine object.
+
+    RAISES:
+        TeradataMlException
+    """
+    host = os.environ.get('TD_HOST')
+    connection_params_from_file = dotenv_values(config_file)
+    if host:
+        connection_params_from_env = {key[3:].lower(): value for key, value in os.environ.items()
+                                      if key.startswith('TD_')}
+        return create_context(**connection_params_from_env)
+    elif connection_params_from_file.get('host'):
+        return create_context(**connection_params_from_file)
+    else:
+        raise TeradataMlException(Messages.get_message(MessageCodes.CONNECTION_PARAMS),
+                                  MessageCodes.MISSING_ARGS)
 
 def _mask_logmech_logdata():
     """
@@ -625,6 +740,8 @@ def set_context(tdsqlengine, temp_database_name=None):
     global temporary_database_name
     global user_specified_connection
     global python_packages_installed
+    global python_version_local
+    global python_version_vantage
     if td_connection is not None:
         # Clearing the internal buffer.
         _InternalBuffer.clean()
@@ -654,6 +771,8 @@ def set_context(tdsqlengine, temp_database_name=None):
     _load_function_aliases()
 
     python_packages_installed = False
+    python_version_vantage = None
+    python_version_local = sys.version.split(" ")[0].strip()
 
     # Initialise Dag
     __initalise_dag()
@@ -687,6 +806,8 @@ def remove_context():
     global td_sqlalchemy_engine
     global user_specified_connection
     global python_packages_installed
+    global python_version_vantage
+    global python_version_local
     global td_user
 
     # Initiate the garbage collection
@@ -706,10 +827,13 @@ def remove_context():
     td_connection = None
     td_sqlalchemy_engine = None
     python_packages_installed = False
+    python_version_local = None
+    python_version_vantage = None
     td_user = None
     configure._current_database_name = None
     configure._database_username = None
     configure.database_version = None
+    configure.indb_install_location = ''
 
     # Closing Dag
     __close_dag()
@@ -1010,19 +1134,21 @@ def _get_host():
     EXAMPLES:
         _get_host()
     """
-    global td_sqlalchemy_engine
-    return td_sqlalchemy_engine.url.host
+    if td_connection is None:
+        return None
+    else:
+        return td_sqlalchemy_engine.url.host
 
 def _get_host_ip():
     """
     DESCRIPTION:
-        Function to return the host IP address.
+        Function to return the host IP address or host name associated with the current context.
 
     PARAMETERS:
         None.
 
     RETURNS:
-        Host IP address.
+        Host IP address or host name associated with the current context.
 
     RAISES:
         None.
@@ -1034,21 +1160,29 @@ def _get_host_ip():
     if td_connection is None:
         return None
     
-    host_ip = _get_host()
+    host = _get_host()
     try:
         # Validate if host_ip is a valid IP address (IPv4 or IPv6)
-        ipaddress.ip_address(host_ip)
+        ipaddress.ip_address(host)
+        return host
     except ValueError:
         # If host is not an IP address, get the IP address by DNS name from _InternalBuffer.
-        host_ip = _InternalBuffer.get('dns_host_ip')
-        if host_ip is None:
-            global td_sqlalchemy_engine
+        dns_host_ip = _InternalBuffer.get('dns_host_ip')
+        if dns_host_ip:
+            return dns_host_ip
+        
+        # If DNS host ip not found, resolve the host name to get the IP address.
+        # If there is issue in resolving the host name, it will proceed with DNS host as it is.
+        try:
             # Get the list of addresses(compatible for both IPv4 and IPv6)
-            addr_info = socket.getaddrinfo(td_sqlalchemy_engine.url.host, None)
+            addr_info = socket.getaddrinfo(host, None)
             # Pick the first address from the list
             host_ip = addr_info[0][4][0]
             # Add the DNS host IP to the _InternalBuffer.
             _InternalBuffer.add(dns_host_ip=host_ip)
+        except socket.gaierror:
+            # Use dns host as it is
+            host_ip = host
     return host_ip
 
 class ContextUtilFuncs():

teradataml/data/amazon_reviews_25.csv

@@ -0,0 +1,26 @@
+"rev_id","aid","rev_name","helpful","rev_text","rating","prodsummary","unixrevtime","revtime"
+A10000012B7CGYKOMPQ4L,"000100039X",Adam,"[0, 0]",Spiritually and mentally inspiring! A book that allows you to question your morals and will help you discover who you really are!,5.00,Wonderful!,1355616000,"12 16, 2012"
+A2KU9IU07LOJS1,"000100039X",Amazon Customer,"[0, 0]",This book has been a classic for many years. It has so much wisdom in it that it can be read numerous times and new things will come out each time. My favorite chapter is the one on children.,5.00,Great classic that everyone should read,1384905600,"11 20, 2013"
+A19N3FCQCLJYUA,"000100039X",Amazon Customer,"[1, 1]","I discovered The Prophet fifty years ago in college and have read it dozens of times since then.  Now, in addition to my beat up hard copy, I have a portable e-copy.  If you were looking for an example of an intelligently designed (pun intended) book of spiritual guidance, this would be it.  It doesn""t care how you picture, name or define "God" or whether you give him a gender and a personality.  It just cuts to the heart of how to live and how to relate to others.  If Jesus and the Buddha teamed up to write a book, it might come out like this.When I first read it, I found some passages difficult to grasp.  Looking back, I now think that it was not because they were hard to understand, but because I did not want to hear gently spoken, intelligently phrased ideas that contradicted my own.  Now, if I could have only one book, this would probably be it.",5.00,A book everyone "should" read,1358899200,"01 23, 2013"
+A5E9TSD20U9PR,"000100039X",April,"[0, 0]","For those who don""t know Gibran, get to know his work. The Prophet is a must read. His outlook on life is truly an inspiring guide on how to approach major life decisions. This is a relationship with a man""s work you won""t regret having.",5.00,Just beautiful.,1377475200,"08 26, 2013"
+A1BM81XB4QHOA3,"000100039X","""Ahoro Blethends """"Seriously""""""","[0, 0]","This book provides a reflection that you can apply to your own life.And, a way for you to try and assess whether you are truly doing the right thing and making the most of your short time on this plane.",5.00,Must Read for Life Afficianados,1390003200,"01 18, 2014"
+A26GKZPS079GFF,"000100039X",Areej,"[2, 3]","I would have to say that this is the best book I""ve ever read.. I could feel every word deep in my heart everytime, of the many times I""ve read it!  I would never get enough of it! its a treasure..",5.00,Touches my heart.. again and.. again...,982972800,"02 24, 2001"
+A1MOSTXNIO5MPJ,"000100039X",Alan Krug,"[0, 0]","I first read THE PROPHET in college back in the 60""s. The book had a revival as did anything metaphysical in the turbulent 60""s. It had a profound effect on me and became a book I always took with me. After graduation I joined the Peace Corps and during stressful training in country (Liberia) at times of illness and the night before I left, this book gave me great comfort. I read it before I married, just before and again after my children were born and again after two near fatal illnesses. I am always amazed that there is a chapter that reaches out to you, grabs you and offers both comfort and hope for the future.Gibran offers timeless insights and love with each word. I think that we as a nation should read AND learn the lessons here. It is definitely a time for thought and reflection this book could guide us through.",5.00,Timeless for every good and bad time in your life.,1317081600,"09 27, 2011"
+A1TT4CY55WLHAR,"000100039X",anonymous,"[0, 0]","I have the 1972 version, bought in 1974. The 1972 version originally had a dust jacket but my dust jacket is long gone. This particular rendition has had many re-printings, for a reason: it""s very popular. The textured paper, old-style typography, and leather cover are better than a plain-old paperback. The size is diminutive, which is perfect for this book.",5.00,"textured paper, old-style typography, and leather cover",1342396800,"07 16, 2012"
+A3FFNE1DR5SI1W,"000100039X",A. Morelli,"[1, 1]","Can""t say enough about Kahlil Gibran""s work among this piece. Everybody in the whole world should read this! There is almost too much to take in, really appreciate and put towards our daily lives. In my opinion, it is just one of the most beautiful literature pieces ever written. Would recommend to as a gift for anyone spiritual/poetic/philosophy/educational piece or just something to enjoy here and there. 5 stars for sure!",5.00,phenomenal piece of literature!,1340755200,"06 27, 2012"
+A1340OFLZBW5NG,"000100039X",Amazon Customer,"[0, 0]",I LOVE this book... his writing seems to just flow from page to page. I get something different from this book each time I read it..,5.00,Perhaps the greatest book that I have ever read,1231977600,"01 15, 2009"
+A29TRDMK51GKZR,"000100039X",Alpine Plume,"[0, 0]","Deep, moving dramatic verses of the heart and soul.Truths of  ancient wisdom from a true and romantic poet.Relevant for all eternity.",5.00,Such Beauty,1383436800,"11 3, 2013"
+A3FI0744PG1WYG,"000100039X","""Always Reading """"tkm""""""","[0, 0]","This is a timeless classic.  Over the years I""ve given it as a gift more times than I can count, and will continue to do so.  Addresses real life issues in a beautiful way and makes us reexamine our own attitude about how we see what happens in our lives.  So easy to read over and over.",5.00,The Prophet,1390953600,"01 29, 2014"
+A2XQ5LZHTD4AFT,"000100039X",Alaturka,"[7, 9]","A timeless classic.  It is a very demanding and assuming title, but Gibran backs it up with some excellent style and content.  If he had the means to publish it a century or two earlier, he could have inspired a new religion.From the mouth of an old man about to sail away to a far away destination, we hear the wisdom of life and all important aspects of it.  It is a messege.  A guide book.  A Sufi sermon. Much is put in perspective without any hint of a dogma.  There is much that hints at his birth place, Lebanon where many of the old prophets walked the Earth and where this book project first germinated most likely.Probably becuase it was written in English originally, the writing flows, it is pleasant to read, and the charcoal drawings of the author decorating the pages is a plus.  I loved the cover.",5.00,A Modern Rumi,1033948800,"10 7, 2002"
+A2LBBQHYLEHM7P,"000100039X","""Amazon Customer """"Full Frontal Nerdity""""""","[0, 0]","An amazing work. Realizing extensive use of Biblical imagery and sentence structure, "The Prophet" by Khalil Gibran is a literary classic. Influencing the Free Love movement of the 1960""s, Gibran""s master work explores themes of love, longing and loss.",5.00,A Modern Classic,1379808000,"09 22, 2013"
+AENNW2G826191,"000100039X",Ashish A,"[1, 4]","Its a thin book, very readable and has interesting 1-2 page thoughts on various entities like anger, children, religion, speech, silence and its COOL.........reading. Ofcourse if one needs to imbibe the thoughts of the  author, it has to be consumed slowly and perhaps revisited but leaves you  pretty heady and clear about certain things.",3.00,Good Read,963446400,"07 13, 2000"
+A2X4HE21JTAL98,"000100039X",Antiquarian,"[3, 5]","Anything I""ve read by Gibran is, in my mind, flawless. This, the most famous of his works, is no exception. It is simple, yet deep; honest and profound; moving and inspirational. Gibran""s work is one of a kind, and can be far more encouraging and moving than any self-help program or therapy or anything like that. The poetic style, the aphorisms, the parables, the almost biblical feel, are all just what over-worked, over-stressed, modern and spiritually starved worldly people need.",5.00,Flawless,1132099200,"11 16, 2005"
+A3V1MKC2BVWY48,"000100039X",Alex Dawson,"[0, 0]","Reading this made my mind feel like a still pool of water, cool and quiet in a mossy grotto.  It""s direct and simple wisdom has a depth of complexity that takes a quiet day to sink in, leaving you at peace.  It is best to set time aside for it, relax, absorb, and let it softly clear your mind.",5.00,This book will bring you peace,1390780800,"01 27, 2014"
+A1KQ80Y692CDOI,"000100039X",Atown,"[2, 9]","I read this about a year ago and can""t recall a great deal of the book. From what I do recall it was like a poem all the way through. While the writing was beautiful, I found it ambiguous and befuddled with meaning that I could not identify with. When Gibran speaks of God, I cannot identify because I have since abandoned those philosophies. It is thus difficult to revisit them in this book. I have the feeling a may have missed something great about this book. Indeed, I pulled wisdom from parts, but rather than go back and read it again, for now, perhaps I will move on to another of the many books out there that are enlightening and worth reading. Someday, I would like to read this again and dig deeper.",2.00,Eloquent,1206057600,"03 21, 2008"
+AUTNO7VDY4H4A,"000100039X",Austin guy,"[0, 0]","Loved this book since first I read it, years gone by.  Purchased this copy for a friend who has not ever read Gibran.",5.00,"A great book, buying it for a friend.",1371427200,"06 17, 2013"
+A2WVHIRDMLM82E,"000100039X",Amazon Customer,"[0, 0]","This book has so much you can take out of it to use in your real life. Amazing, and one of my favorite reads of all time.",5.00,Amazing,1394928000,"03 16, 2014"
+A2I35JB67U20C0,"000100039X",Amazon Customer,"[0, 0]","When I was in college in the 70""s this book had a revival and I did not read it then.  Recently a friend (who is 90) and I were talking about work and she said: "work is love made visible," and told me it was from the Prophet.  I though that was so beautiful I got the book and was not disappointed.You see, if you have ideas and you do not realize them, then they are nothing, and if your ideas do not come from love and joy, then they are bitter and what they produce will be bitter, but if they flow from love and joy, then their realization will be love and joy, thus work is love made visible.He says it WAY better that I do and says much more in just a few paragraphs.  But as you can see from the example, what he says is not religious, but positive ways of looking at things.  Since I rediscoverd the book I have given away many copies and everyone I have given it to sincerly thanked me.",5.00,Everyone should have this book,983318400,"02 28, 2001"
+A12387207U8U24,"000100039X",Alex,"[0, 0]","As you read, Gibran""s poetry brings spiritual and visual beauty to life within you. Gibran is justly famous for rich metaphors that brilliantly highlight the pursuit of Truth and Goodness amidst all the darkness and light of human nature.",5.00,Graet Work,1206662400,"03 28, 2008"
+A2S166WSCFIFP5,"000100039X","""adead_poet@hotmail.com """"adead_poet@hotmail.com""""""","[0, 2]","This is one my must have books. It is a masterpiece of spirituality. I""ll be the first to admit, its literary quality isn""t much. It is rather simplistically written, but the message behind it is so powerful that you have to read it. It will take you to enlightenment.",5.00,close to god,1071100800,"12 11, 2003"
+A27ZH1AQORJ1L,"000100039X","""anybody else or """"amanuet""""""","[3, 3]","This book is everything that is simple, delicate, true, and beautiful.I have read few books so touching and enlightening; "The Prophet" is a true masterpiece that has that feeling of ancient wisdom in it. The wisdom of the text is gentle, yet insistent, it lets you understand things you""ve always known.My feelings defy description.",5.00,Enchanting,1066003200,"10 13, 2003"
+ARDQ9KNB8K22N,"000100039X",Anwar,"[1, 1]","Cool book, I really like the quality of the production.  Black clothbound with gold embossing and nice paper, looks to be cotton rag.  I am proud to include it in my collection.  Intrigueing story full of the sort of passion that does not cloud the mind or divert truth but instead is revealing and living. Very readable, the stories are short and highly economic so that one is likely to read for only a minute or two until something insightful is revealed. It is full of timeless truths which are of lasting value to the reader and inform life.  The language and delivery is familiar yet of a quality that is penetrating in a similar way as it might be listening to a surmon by Jesus.  I give it 5 stars. I considered giving only 4 stars because the format is fairly predictable but I have decided that this quality is actually a strength as one can pick up the book at any point and continue until the end...and then start over for that matter.  Excellent for busy people or commuters!",5.00,"""""""The Prophet"""" is cool""",1329264000,"02 15, 2012"

teradataml/data/byom_example.json

@@ -14,5 +14,16 @@
     "petal_length" : "float",
     "petal_width" : "float",
     "species": "integer"
+  },
+  "amazon_reviews_25": {
+    "rev_id": "VARCHAR(64000)",
+    "aid": "VARCHAR(64000)",
+    "rev_name": "VARCHAR(64000)",
+    "helpful": "VARCHAR(64000)",
+    "rev_text": "VARCHAR(64000)",
+    "rating": "DECIMAL(10,2)",
+    "prodsummary": "VARCHAR(64000)",
+    "unixrevtime": "BIGINT",
+    "revtime": "VARCHAR(64000)"
   }
 }

teradataml/data/dataframe_example.json

@@ -152,6 +152,22 @@
     "item" : "varchar(20)",
     "sku" : "integer",
     "category" : "varchar(20)"
-
-   }
+   },
+  "medical_readings": {
+    "patient_id":  "BIGINT",
+    "record_timestamp":  "timestamp",
+    "glucose":  "BIGINT",
+    "blood_pressure":  "BIGINT",
+    "insulin":  "BIGINT",
+    "diabetes_pedigree_function":  "FLOAT",
+    "outcome": "BIGINT"
+  },
+  "patient_profile": {
+    "patient_id":  "BIGINT",
+    "record_timestamp":  "timestamp",
+    "pregnancies":  "BIGINT",
+    "age":  "BIGINT",
+    "bmi":  "FLOAT",
+    "skin_thickness":  "FLOAT"
+  }
 }

teradataml/data/docs/byom/docs/DataRobotPredict.py

@@ -23,7 +23,7 @@ def DataRobotPredict(modeldata=None, newdata=None, accumulate=None, model_output
             Required Argument.
             Specifies the name(s) of input teradataml DataFrame column(s) to
             copy to the output.
-            Types: str OR list of Strings (str)
+            Types: str OR list of Strings (str) OR Feature OR list of Features
 
         model_output_fields:
             Optional Argument.
@@ -108,7 +108,7 @@ def DataRobotPredict(modeldata=None, newdata=None, accumulate=None, model_output
             for each argument that accepts teradataml DataFrame as
             input and can be accessed as:
                 * "<input_data_arg_name>_partition_column" accepts str or
-                  list of str (Strings)
+                  list of str (Strings) or PartitionKind
                 * "<input_data_arg_name>_hash_column" accepts str or list
                   of str (Strings)
                 * "<input_data_arg_name>_order_column" accepts str or list

teradataml/data/docs/byom/docs/DataikuPredict.py

@@ -24,7 +24,7 @@ def DataikuPredict(modeldata=None, newdata=None, accumulate=None, model_output_f
             Specifies the name(s) of input teradataml DataFrame column(s) to
             copy to the output. By default, the function copies all input
             teradataml DataFrame columns to the output.
-            Types: str OR list of Strings (str)
+            Types: str OR list of Strings (str) OR Feature OR list of Features
 
         model_output_fields:
             Optional Argument.
@@ -86,6 +86,45 @@ def DataikuPredict(modeldata=None, newdata=None, accumulate=None, model_output_f
             Default Value: False
             Types: bool
 
+        **generic_arguments:
+            Specifies the generic keyword arguments SQLE functions accept. Below
+            are the generic keyword arguments:
+                persist:
+                    Optional Argument.
+                    Specifies whether to persist the results of the
+                    function in a table or not. When set to True,
+                    results are persisted in a table; otherwise,
+                    results are garbage collected at the end of the
+                    session.
+                    Default Value: False
+                    Types: bool
+
+                volatile:
+                    Optional Argument.
+                    Specifies whether to put the results of the
+                    function in a volatile table or not. When set to
+                    True, results are stored in a volatile table,
+                    otherwise not.
+                    Default Value: False
+                    Types: bool
+
+            Function allows the user to partition, hash, order or local
+            order the input data. These generic arguments are available
+            for each argument that accepts teradataml DataFrame as
+            input and can be accessed as:
+                * "<input_data_arg_name>_partition_column" accepts str or
+                  list of str (Strings) or PartitionKind
+                * "<input_data_arg_name>_hash_column" accepts str or list
+                  of str (Strings)
+                * "<input_data_arg_name>_order_column" accepts str or list
+                  of str (Strings)
+                * "local_order_<input_data_arg_name>" accepts boolean
+            Note:
+                These generic arguments are supported by teradataml if
+                the underlying SQL Engine function supports, else an
+                exception is raised.
+
+
     RETURNS:
         Instance of DataikuPredict.
         Output teradataml DataFrame can be accessed using attribute

teradataml/data/docs/byom/docs/H2OPredict.py

@@ -38,7 +38,7 @@ def H2OPredict(modeldata=None, newdata=None, accumulate=None, model_output_field
             Required Argument.
             Specifies the name(s) of input teradataml DataFrame column(s)
             to copy to the output DataFrame.
-            Types: str OR list of Strings (str)
+            Types: str OR list of Strings (str) OR Feature OR list of Features
 
         model_output_fields:
             Optional Argument.
@@ -147,7 +147,7 @@ def H2OPredict(modeldata=None, newdata=None, accumulate=None, model_output_field
             for each argument that accepts teradataml DataFrame as
             input and can be accessed as:
                 * "<input_data_arg_name>_partition_column" accepts str or
-                  list of str (Strings)
+                  list of str (Strings) or PartitionKind
                 * "<input_data_arg_name>_hash_column" accepts str or list
                   of str (Strings)
                 * "<input_data_arg_name>_order_column" accepts str or list