Fixed typing
This commit is contained in:
parent
76f24d4eb0
commit
30c2938d29
@ -1,6 +1,7 @@
|
|||||||
from enum import Enum
|
from enum import Enum
|
||||||
|
|
||||||
class SpecialToken(str, Enum):
|
|
||||||
|
class SpecialToken(Enum):
|
||||||
# (Enum, str) -> throws an error
|
# (Enum, str) -> throws an error
|
||||||
START_TRIPLE_LIST = "<SOTL>"
|
START_TRIPLE_LIST = "<SOTL>"
|
||||||
START_TRIPLE = "<SOT>"
|
START_TRIPLE = "<SOT>"
|
||||||
@ -18,4 +19,3 @@ class SpecialToken(str, Enum):
|
|||||||
MASK = "<MASK>"
|
MASK = "<MASK>"
|
||||||
|
|
||||||
# BPE Training:
|
# BPE Training:
|
||||||
|
|
||||||
Loading…
x
Reference in New Issue
Block a user