########################################################################################################################
#                                                ENVIRONMENT VARIABLES                                                 #
########################################################################################################################
NUM_GPU = "NUM_GPU"
SAGEMAKER_INFERENCE_OUTPUT = "SAGEMAKER_INFERENCE_OUTPUT"

########################################################################################################################
#                                                OUTPUT CONSTANTS                                                      #
########################################################################################################################
PROBABILITY = "probability"
PROBABILITIES = "probabilities"
PREDICTED_LABEL = "predicted_label"
LABELS = "labels"

########################################################################################################################
#                                                DATA FORMAT CONSTANTS                                                 #
########################################################################################################################
BYTE_ARRAY_FORMAT = "application/x-image"
JPEG_FORMAT = "image/jpeg"
PNG_FORMAT = "image/png"
JSON_FORMAT = "application/json"
CSV_FORMAT = "text/csv"
COMMA_DELIMITER = ","
BRACKET_FORMATTER = '"{}"'
ALLOWED_INPUT_FORMATS = [BYTE_ARRAY_FORMAT, JPEG_FORMAT, PNG_FORMAT]
ALLOWED_OUTPUT_FORMATS = [JSON_FORMAT, CSV_FORMAT]

########################################################################################################################
#                                                INFERENCE DATA CONSTANTS                                              #
########################################################################################################################
IMAGE_COLUMN_NAME = "image"