teradataml 17.20.0.6__py3-none-any.whl → 20.0.0.0__py3-none-any.whl

teradataml/common/messagecodes.py

@@ -136,7 +136,6 @@ class ErrorInfoCodes(Enum):
     MODEL_ALREADY_EXISTS = 'TDML_2200'
     MODEL_NOT_FOUND = 'TDML_2201'
     MODEL_WITH_SEARCH_CRITERION_NOT_FOUND = 'TDML_2202'
-    MODEL_CATALOGING_TABLE_DOES_EXIST = 'TDML_2203'
     MODEL_CATALOGING_OPERATION_FAILED = 'TDML_2204'
     FUNCTION_JSON_MISSING = 'TDML_2205'
     UNKNOWN_MODEL_ENGINE = 'TDML_2206'
@@ -180,18 +179,6 @@ class ErrorInfoCodes(Enum):
     EITHER_FUNCTION_OR_ARGS = 'TDML_2313'
     INVALID_LIST_LENGTH = 'TDML_2314'
 
-    # Sandbox related Error Codes
-    SANDBOX_CONTAINER_ERROR = 'TDML_2400'
-    SANDBOX_CONNECTION_ERROR = 'TDML_2401'
-    SANDBOX_SCRIPT_ERROR = 'TDML_2402'
-    SANDBOX_QUERY_ERROR = 'TDML_2403'
-    SANDBOX_CONTAINER_NOT_RUNNING = 'TDML_2404'
-    SANDBOX_IMAGE_NOT_FOUND = 'TDML_2405'
-    SANDBOX_CONTAINER_NOT_FOUND = 'TDML_2406'
-    CONTAINER_STARTED_BY_TERADATAML_EXISTS = 'TDML_2407'
-    SANDBOX_CONTAINER_EXISTS = 'TDML_2408'
-    SANDBOX_SKIP_IMAGE_LOAD = 'TDML_2409'
-    SANDBOX_CONTAINER_CAN_NOT_BE_STARTED = 'TDML_2410'
     IMPORT_PYTHON_PACKAGE = 'TDML_2414'
 
     # Script local run Error codes
@@ -221,6 +208,17 @@ class ErrorInfoCodes(Enum):
     INVALID_DICT_KEY_VALUE_LENGTH = 'TDML_2525'
     DUPLICATE_DICT_KEYS_NAMES = 'TDML_2526'
 
+    # Opensource Error codes starting from 2531 - Reserved till 2550.
+    MODEL_NOT_FITTED = 'TDML_2531'
+    DFS_NO_COMMON_PARENT = 'TDML_2532'
+    NODE_NOT_GIVEN_TYPE = 'TDML_2534'
+    ARGS_WITH_SAME_COLUMNS = 'TDML_2535'
+    PARTITIONING_COLS_DIFFERENT = 'TDML_2536'
+    PARTITIONING_COLS_IN_FEATURE_COLS = 'TDML_2537'
+    PARTITION_VALUES_NOT_MATCHING = 'TDML_2538'
+    PARTITION_IN_BOTH_FIT_AND_PREDICT = 'TDML_2539'
+    INVALID_PARTITIONING_COLS = 'TDML_2540'
+
 class MessageCodes(Enum):
     """
     MessageCodes contains all the messages that are displayed to the user which are informational
@@ -325,7 +323,6 @@ class MessageCodes(Enum):
     MODEL_ALREADY_EXISTS            = "Model with name '{}' already exists."
     MODEL_WITH_SEARCH_CRITERION_NOT_FOUND = "Model with the given search criterion not found."
     MODEL_NOT_FOUND                 = "Model '{}' not found{}."
-    MODEL_CATALOGING_TABLE_DOES_EXIST = "Model Catalog not set up."
     MODEL_CATALOGING_OPERATION_FAILED = "Failed to {} the model. {}"
     UNKNOWN_MODEL_ENGINE            = "Unable to find model generating engine for model of type {}"
     FUNCTION_JSON_MISSING           = "Failed to load JSON file for function: '{}', engine: '{}'"
@@ -374,7 +371,6 @@ class MessageCodes(Enum):
     INVALID_LENGTH_STRING_ARG       = "Length of the string passed to '{}' argument should be {}."
     INVALID_LENGTH_ARG              = "Length of the {} passed to '{}' argument should be {}."
     LIST_SELECT_NONE_OR_EMPTY = "{} specified contains an empty string/list or an invalid index (such as '', None, [None] or ['None'])."
-    SANDBOX_CONTAINER_ERROR         = "Sandbox container error: {}"
     DATAFRAME_LIMIT_ERROR           = "Number of rows in the Dataframe is greater than {}. Either increase the limit on the number of rows by setting {} or create a dataframe with less than or equal to {} rows."
     SPECIFY_AT_LEAST_ONE_ARG        = "At least one of the '{}' or '{}' arguments must be specified."
     NOT_ALLOWED_VALUES   = "'{}' is not allowed for argument '{}'."
@@ -386,33 +382,13 @@ class MessageCodes(Enum):
     MIXED_TYPES_IN_COLUMN_RANGE     = "Column range specified for the argument '{}' has mix of index and column name. " \
                                       "Provide either indices or column names."
     SCRIPT_LOCAL_RUN_ERROR = "Error occurred while running user script locally: {}"
-    SANDBOX_CONNECTION_ERROR = "Connection to Vantage failed: {}"
-    SANDBOX_SCRIPT_ERROR = "Error occurred while running user script in sandbox: {}"
-    SANDBOX_QUERY_ERROR = "Error accessing data from table: {}"
     STORED_PROCEDURE_FAILED         = "Failed to execute function '{}'.\n{}"
     NO_ENVIRONMENT_FOUND            = "No {} environment(s) found."
     UNSUPPORTED_FILE_EXTENSION      = "Unsupported file extension specified. Supported file extensions is/are {}."
     FILE_EMPTY               = "Input file {} is empty."
 
-    SANDBOX_CONTAINER_EXISTS = "Sandbox container already exists. Please cleanup or use existing."
-    SANDBOX_IMAGE_NOT_FOUND = "Sandbox image not found. Please load the image first."
-    SANDBOX_CONTAINER_NOT_FOUND = "Sandbox container not found. Please run setup_sandbox_env() first " \
-                                  "or set configure.sandbox_container_id."
-    SANDBOX_CONTAINER_NOT_RUNNING = "Container {} is not running. Please start the container."
-    SANDBOX_SKIP_IMAGE_LOAD = "Skipped image loading since sandbox image {} already exists on the system."
     PYTHON_NOT_INSTALLED = "Python is not installed on Vantage. " \
                            "Please install Python interpreter and add-on packages on Vantage."
-    CONTAINER_STARTED_BY_TERADATAML_EXISTS = "Container started by teradataml exists. " \
-                                             "There can only be one running container " \
-                                             "started by teradataml using setup_sandbox_env(). " \
-                                             "In order to start a new container please run " \
-                                             "cleanup_sandbox_env() first. Alternatively, " \
-                                             "you can choose to start a container from " \
-                                             "outside teradataml."
-    SANDBOX_CONTAINER_CAN_NOT_BE_STARTED = "STO sandbox image name {} does not match with " \
-                                           "the image. Container cannot be started. " \
-                                           "Removing already loaded image. Please re-run " \
-                                           "with correct image name."
     IMPORT_PYTHON_PACKAGE = "Module '{}' not found. Install '{}' before running {}()."
     INT_ARGUMENT_COMPARISON = "Argument '{}' must be {} to argument '{}'"
     EXECUTION_FAILED = "Failed to {}. {}"
@@ -428,4 +404,19 @@ class MessageCodes(Enum):
                             "the 'user_field_count'."
     RESERVED_KEYWORD = "'{}' is a Teradata reserved keyword."
     INVALID_LIST_LENGTH = "Length of the list passed to argument '{}' should be of size {}."
-    FUNC_EXECUTION_FAILED = "Failed to execute {}. {}"
+    FUNC_EXECUTION_FAILED = "Failed to execute {}. {}"
+    MODEL_NOT_FITTED = "Model should be fitted before running {}() or getting attributes."
+    DFS_NO_COMMON_PARENT = "DataFrames passed must be derived from same DataFrame, "\
+                            "i.e., their parent DataFrame must be same."
+    NODE_NOT_GIVEN_TYPE = "The DataFrame passed in '{}' argument is not a DataFrame derived using {}() API."
+    ARGS_WITH_SAME_COLUMNS = "Column(s)/DataFrame columns(s) in '{}' and '{}' arguments "\
+                                "should{} be same."
+    PARTITIONING_COLS_DIFFERENT = "Model is fitted using 'partition_columns' but " \
+                "'partition_columns' are not passed. Partition columns should be same as fit()'s "\
+                "'partition_columns' or pass 'partition_columns' to the function. In either case "\
+                "they should be present in '{}' DataFrame."
+    PARTITIONING_COLS_IN_FEATURE_COLS = "Columns in '{}' argument should not be part of"\
+                                        " feature columns."
+    PARTITION_VALUES_NOT_MATCHING = "Values in training and test data partition columns should be same."
+    PARTITION_IN_BOTH_FIT_AND_PREDICT = "Use \"partition_columns\" only if model is fitted with partition_column(s)."
+    INVALID_PARTITIONING_COLS = "Provided partition_column(s) '{}' is/are not present in parent of '{}' DataFrame(s)."

teradataml/common/messages.py

@@ -107,7 +107,6 @@ class Messages():
             [ErrorInfoCodes.MODEL_ALREADY_EXISTS, MessageCodes.MODEL_ALREADY_EXISTS],
             [ErrorInfoCodes.MODEL_NOT_FOUND, MessageCodes.MODEL_NOT_FOUND],
             [ErrorInfoCodes.MODEL_WITH_SEARCH_CRITERION_NOT_FOUND, MessageCodes.MODEL_WITH_SEARCH_CRITERION_NOT_FOUND],
-            [ErrorInfoCodes.MODEL_CATALOGING_TABLE_DOES_EXIST, MessageCodes.MODEL_CATALOGING_TABLE_DOES_EXIST],
             [ErrorInfoCodes.TABLE_DOES_NOT_EXIST, MessageCodes.TABLE_DOES_NOT_EXIST],
             [ErrorInfoCodes.EMPTY_FILE, MessageCodes.EMPTY_FILE],
             [ErrorInfoCodes.MODEL_CATALOGING_OPERATION_FAILED, MessageCodes.MODEL_CATALOGING_OPERATION_FAILED],
@@ -153,7 +152,6 @@ class Messages():
             [ErrorInfoCodes.INVALID_LENGTH_STRING_ARG, MessageCodes.INVALID_LENGTH_STRING_ARG],
             [ErrorInfoCodes.INVALID_LENGTH_ARG, MessageCodes.INVALID_LENGTH_ARG],
             [ErrorInfoCodes.LIST_SELECT_NONE_OR_EMPTY, MessageCodes.LIST_SELECT_NONE_OR_EMPTY],
-            [ErrorInfoCodes.SANDBOX_CONTAINER_ERROR, MessageCodes.SANDBOX_CONTAINER_ERROR],
             [ErrorInfoCodes.DATAFRAME_LIMIT_ERROR, MessageCodes.DATAFRAME_LIMIT_ERROR],
             [ErrorInfoCodes.SPECIFY_AT_LEAST_ONE_ARG, MessageCodes.SPECIFY_AT_LEAST_ONE_ARG],
             [ErrorInfoCodes.NOT_ALLOWED_VALUES, MessageCodes.NOT_ALLOWED_VALUES],
@@ -162,24 +160,13 @@ class Messages():
             [ErrorInfoCodes.UNKNOWN_ARGUMENT, MessageCodes.UNKNOWN_ARGUMENT],
             [ErrorInfoCodes.CANNOT_USE_TOGETHER_WITH, MessageCodes.CANNOT_USE_TOGETHER_WITH],
             [ErrorInfoCodes.SCRIPT_LOCAL_RUN_ERROR, MessageCodes.SCRIPT_LOCAL_RUN_ERROR],
-            [ErrorInfoCodes.SANDBOX_CONNECTION_ERROR,MessageCodes.SANDBOX_CONNECTION_ERROR],
-            [ErrorInfoCodes.SANDBOX_QUERY_ERROR, MessageCodes.SANDBOX_QUERY_ERROR],
-            [ErrorInfoCodes.SANDBOX_SCRIPT_ERROR, MessageCodes.SANDBOX_SCRIPT_ERROR],
             [ErrorInfoCodes.INVALID_COLUMN_RANGE_FORMAT, MessageCodes.INVALID_COLUMN_RANGE_FORMAT],
             [ErrorInfoCodes.MIXED_TYPES_IN_COLUMN_RANGE, MessageCodes.MIXED_TYPES_IN_COLUMN_RANGE],
             [ErrorInfoCodes.STORED_PROCEDURE_FAILED, MessageCodes.STORED_PROCEDURE_FAILED],
             [ErrorInfoCodes.NO_ENVIRONMENT_FOUND, MessageCodes.NO_ENVIRONMENT_FOUND],
             [ErrorInfoCodes.UNSUPPORTED_FILE_EXTENSION, MessageCodes.UNSUPPORTED_FILE_EXTENSION],
             [ErrorInfoCodes.FILE_EMPTY, MessageCodes.FILE_EMPTY],
-            [ErrorInfoCodes.SANDBOX_CONTAINER_EXISTS, MessageCodes.SANDBOX_CONTAINER_EXISTS],
-            [ErrorInfoCodes.SANDBOX_IMAGE_NOT_FOUND, MessageCodes.SANDBOX_IMAGE_NOT_FOUND],
-            [ErrorInfoCodes.SANDBOX_CONTAINER_NOT_FOUND, MessageCodes.SANDBOX_CONTAINER_NOT_FOUND],
-            [ErrorInfoCodes.SANDBOX_SKIP_IMAGE_LOAD, MessageCodes.SANDBOX_SKIP_IMAGE_LOAD],
             [ErrorInfoCodes.PYTHON_NOT_INSTALLED, MessageCodes.PYTHON_NOT_INSTALLED],
-            [ErrorInfoCodes.CONTAINER_STARTED_BY_TERADATAML_EXISTS,
-             MessageCodes.CONTAINER_STARTED_BY_TERADATAML_EXISTS],
-            [ErrorInfoCodes.SANDBOX_CONTAINER_NOT_RUNNING, MessageCodes.SANDBOX_CONTAINER_NOT_RUNNING],
-            [ErrorInfoCodes.SANDBOX_CONTAINER_CAN_NOT_BE_STARTED, MessageCodes.SANDBOX_CONTAINER_CAN_NOT_BE_STARTED],
             [ErrorInfoCodes.INT_ARGUMENT_COMPARISON, MessageCodes.INT_ARGUMENT_COMPARISON],
             [ErrorInfoCodes.EXECUTION_FAILED, MessageCodes.EXECUTION_FAILED],
             [ErrorInfoCodes.INVALID_COLUMN_DATATYPE, MessageCodes.INVALID_COLUMN_DATATYPE],
@@ -192,13 +179,22 @@ class Messages():
             [ErrorInfoCodes.RESERVED_KEYWORD, MessageCodes.RESERVED_KEYWORD],
             [ErrorInfoCodes.INVALID_LIST_LENGTH, MessageCodes.INVALID_LIST_LENGTH],
             [ErrorInfoCodes.EXECUTION_FAILED, MessageCodes.FUNC_EXECUTION_FAILED],
-            [ErrorInfoCodes.IMPORT_PYTHON_PACKAGE, MessageCodes.IMPORT_PYTHON_PACKAGE]
+            [ErrorInfoCodes.IMPORT_PYTHON_PACKAGE, MessageCodes.IMPORT_PYTHON_PACKAGE],
+            [ErrorInfoCodes.MODEL_NOT_FITTED, MessageCodes.MODEL_NOT_FITTED],
+            [ErrorInfoCodes.DFS_NO_COMMON_PARENT, MessageCodes.DFS_NO_COMMON_PARENT],
+            [ErrorInfoCodes.NODE_NOT_GIVEN_TYPE, MessageCodes.NODE_NOT_GIVEN_TYPE],
+            [ErrorInfoCodes.ARGS_WITH_SAME_COLUMNS, MessageCodes.ARGS_WITH_SAME_COLUMNS],
+            [ErrorInfoCodes.PARTITIONING_COLS_DIFFERENT, MessageCodes.PARTITIONING_COLS_DIFFERENT],
+            [ErrorInfoCodes.PARTITIONING_COLS_IN_FEATURE_COLS, MessageCodes.PARTITIONING_COLS_IN_FEATURE_COLS],
+            [ErrorInfoCodes.PARTITION_VALUES_NOT_MATCHING, MessageCodes.PARTITION_VALUES_NOT_MATCHING],
+            [ErrorInfoCodes.PARTITION_IN_BOTH_FIT_AND_PREDICT, MessageCodes.PARTITION_IN_BOTH_FIT_AND_PREDICT],
+            [ErrorInfoCodes.INVALID_PARTITIONING_COLS, MessageCodes.INVALID_PARTITIONING_COLS]
     ]
 
     @staticmethod
     def get_message(messagecode, *variables, **kwargs):
         """
-        Generate a message associated with standard message and error code .
+        Generate a message associated with standard message and error code.
 
         PARAMETERS:
             messagecode(Required)  - Message  to be returned to the user when needed to be raised based on \