Skip to content

Commit 306ef10

Browse files
committed
Fix preprocessing
1 parent cdb3191 commit 306ef10

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

sota_extractor2/models/structure/ulmfit.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,3 +13,6 @@ def _fix_sp_processor(self, sp_path, sp_model, sp_vocab):
1313
if isinstance(processor, SPProcessor):
1414
processor.sp_model = sp_path / sp_model
1515
processor.sp_vocab = sp_path / sp_vocab
16+
17+
#todo: see why it wasn't set on save
18+
processor.mark_fields = True

0 commit comments

Comments
 (0)