import os

model_type = os.environ['MODEL_TYPE']
assert model_type == 'Roberta' or model_type == 'Bert' or model_type == 'Albert'
SEP_TOKEN = '[SEP]' if model_type == 'Bert' or model_type == 'Albert' else '</s>'
CLS_TOKEN = '[CLS]' if model_type == 'Bert' or model_type == 'Albert' else '<s>'
NODE_SEP_TOKEN = '[unused1]' if model_type == 'Bert' or model_type == 'Albert' else '|'
QUESTION_SEP_TOKEN = '[unused2]' if model_type == 'Bert' or model_type == 'Albert' else '^'
tokenizer = None
eval_test = True
model_args = None
rgcn_relation_nums = -1
node_intervals_padding_id = -1
node_interval_padding_len = 50
extension_padding_value = -1
extension_padding_len = 10
seed = 42
extension_threshold = 0.5
truncate_edges_num = int(os.environ['truncate_edges_num'])
truncate_nodes_num = int(os.environ['truncate_nodes_num'])
max_edge_num = 101
max_node_num = 50
