Fix bug

nlpodyssey · Nov 8, 2023 · f0c0f0a · f0c0f0a
1 parent 31d2aba
commit f0c0f0a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/tasks/tokenclassification/utils.go b/pkg/tasks/tokenclassification/utils.go
@@ -64,7 +64,7 @@ func stripPrefix(label string) string {
 	switch {
 	case label == "O": // outside
 		return ""
-	case len(label) > 2:
+	case len(label) > 2 && label[1] == '-':
 		return label[2:]
 	default:
 		return label