"""config_fields.py - constants for fields in config YAML."""

# TODO: define fields expected to be found in config YAML
# based on the neat schema.

# TODO: clean up some redundancies as marked below

APPLYTRAINEDMODELSCONTAINER = "ApplyTrainedModelsContainer"
BERT_COLUMNS = "bert_columns"
BERT_PARAMS = "bert_params"
CLASSIFIERCONTAINER = "ClassifierContainer"
CLASSIFIERS = "classifiers"
CLASSIFIER_ID = "classifier_id"
CLASSIFIER_NAME = "classifier_name"
CUTOFF = "cutoff"
DESC = "desc"
DESTINATION = "destination"
EDGE_METHOD = "edge_method"
EDGE_PATH = "edge_path"
EMBEDDINGSCONFIG = "EmbeddingsConfig"
EMBEDDINGS_FILE = "embeddings_file"  # redundant
EMBEDDING_FILE = "embedding_file"  # redundant
EMBEDDING_HISTORY_OUTFILE = "embedding_history_outfile"
EMBEDDING_OUTFILE = "embedding_outfile"
EVALUATION_DATA = "evaluation_data"
EXTRA_ARGS = "extra_args"
FILENAME = "filename"
FILES = "files"
GRAPH = "graph"
GRAPHDATACONFIGURATION = "GraphDataConfiguration"
HISTORY_FILENAME = "history_filename"
MAIN_GRAPH_ARGS = "main_graph_args"
MODEL = "model"
MODELS = "models"
MODEL_ID = "model_id"
NODES_COLUMN = "nodes_column"
NODE_COLUMNS = "node_columns"
NODE_EMBEDDING_PARAMS = "node_embedding_params"  # redundant
NODE_EMBEDDINGS_PARAMS = "node_embeddings_params"  # redundant
NODE_LIST_NODE_TYPES_COLUMN = "node_list_node_types_column"
NODE_PATH = "node_path"
NODE_TYPES = "node_types"
OUTFILE = "outfile"  # redundant
OUTPUT_FILE = "output_file"  # redundant
PATH = "path"
POS_EDGE_FILEPATH = "pos_edge_filepath"
S3_BUCKET = "s3_bucket"
S3_BUCKET_DIR = "s3_bucket_dir"
SOURCE = "source"
SOURCE_DATA = "source_data"
TARGET = "Target"
TARGET_PATH = "target_path"
TRAIN_DATA = "train_data"
TSNE_FILENAME = "tsne_filename"  # redundant
TSNE_OUTFILE = "tsne_outfile"  # redundant
UPLOAD = "Upload"
VALID_DATA = "valid_data"
