mirror of https://github.com/open-mmlab/mmocr.git
[Fix] Wildreceipt tests (#1546)
parent
e067ddea23
commit
3433c8cba4
mmocr/datasets/preparers/dumpers
|
@ -1,6 +1,6 @@
|
|||
# Copyright (c) OpenMMLab. All rights reserved.
|
||||
import os.path as osp
|
||||
from typing import Dict, List
|
||||
from typing import Dict, List, Optional
|
||||
|
||||
import mmengine
|
||||
|
||||
|
@ -56,9 +56,8 @@ class JsonDumper:
|
|||
@DATA_DUMPERS.register_module()
|
||||
class WildreceiptOpensetDumper:
|
||||
|
||||
def __init__(self, task: str, dataset_name: str) -> None:
|
||||
def __init__(self, task: str, dataset_name: Optional[str] = None) -> None:
|
||||
self.task = task
|
||||
self.dataset_name = dataset_name
|
||||
|
||||
def dump(self, data: List, data_root: str, split: str) -> str:
|
||||
"""Dump data to txt file.
|
||||
|
|
Loading…
Reference in New Issue