Fixed typing
This commit is contained in:
@@ -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:
|
||||||
|
|
||||||
Reference in New Issue
Block a user