nl4opt-subtask2-baseline/constants.py at main · nl4opt/nl4opt-subtask2-baseline · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
START_OF_TEMPLATE = '<DECLARATION>'
END_OF_TEMPLATE = '</DECLARATION>'
START_OF_VAR = '<VAR>'
END_OF_VAR = '</VAR>'
START_OF_PARAM = '<PARAM>'
END_OF_PARAM = '</PARAM>'
START_OF_LIMIT= '<LIMIT>'
END_OF_LIMIT = '</LIMIT>'
START_OF_OBJ_DIR = '<OBJ_DIR>'
END_OF_OBJ_DIR = '</OBJ_DIR>'
START_OF_OBJ_NAME = '<OBJ_NAME>'
END_OF_OBJ_NAME = '</OBJ_NAME>'
START_OF_CONST_DIR = '<CONST_DIR>'
END_OF_CONST_DIR = '</CONST_DIR>'
START_OF_CONST_TYPE = '<CONST_TYPE>'
END_OF_CONST_TYPE = '</CONST_TYPE>'
START_OF_OPERATOR = '<OPERATOR>'
END_OF_OPERATOR = '</OPERATOR>'

IS_TOKEN = '[is]'
TIMES_TOKEN = '[times]'
PLUS_TOKEN = '[plus]'
FOR_TOKEN = '[for]'

ONE_TOKEN = 'ONE'

TYPE_UPPER_BOUND = "[UPPER_BOUND]"
TYPE_LOWER_BOUND = "[LOWER_BOUND]"
TYPE_SUM_CONST = "[SUM_CONSTRAINT]"
TYPE_LINEAR_CONST = "[LINEAR_CONSTRAINT]"
TYPE_RATIO_CONST = "[RATIO_CONSTRAINT]"
TYPE_XBY_CONST = "[XBY_CONSTRAINT]"
TYPE_XY_CONST = "[XY_CONSTRAINT]"

SPECIAL_TOKENS = [
    START_OF_TEMPLATE, END_OF_TEMPLATE,
    START_OF_VAR, END_OF_VAR,
    START_OF_LIMIT, END_OF_LIMIT,
    START_OF_OBJ_DIR, END_OF_OBJ_DIR,
    START_OF_CONST_DIR, END_OF_CONST_DIR,
    START_OF_CONST_TYPE, END_OF_CONST_TYPE,
    START_OF_OPERATOR, END_OF_OPERATOR,
    IS_TOKEN,
    TIMES_TOKEN,
    PLUS_TOKEN,
    FOR_TOKEN,
    TYPE_UPPER_BOUND,
    TYPE_LOWER_BOUND,
    TYPE_SUM_CONST,
    TYPE_LINEAR_CONST,
    TYPE_RATIO_CONST,
    TYPE_XBY_CONST,
    TYPE_XY_CONST,
    ONE_TOKEN
]