Joshua Lochner commited on
Commit
fca2a61
1 Parent(s): bb58e90

Move `CATGEGORY_OPTIONS` to shared

Browse files
Files changed (1) hide show
  1. src/shared.py +6 -0
src/shared.py CHANGED
@@ -9,6 +9,12 @@ from dataclasses import dataclass, field
9
  from enum import Enum
10
 
11
 
 
 
 
 
 
 
12
  START_SEGMENT_TEMPLATE = 'START_{}_TOKEN'
13
  END_SEGMENT_TEMPLATE = 'END_{}_TOKEN'
14
 
 
9
  from enum import Enum
10
 
11
 
12
+ CATGEGORY_OPTIONS = {
13
+ 'SPONSOR': 'Sponsor',
14
+ 'SELFPROMO': 'Self/unpaid promo',
15
+ 'INTERACTION': 'Interaction reminder',
16
+ }
17
+
18
  START_SEGMENT_TEMPLATE = 'START_{}_TOKEN'
19
  END_SEGMENT_TEMPLATE = 'END_{}_TOKEN'
20