[Fix] Wildreceipt tests ()

pull/1547/head
Tong Gao 2022-11-17 10:39:11 +08:00 committed by GitHub
parent e067ddea23
commit 3433c8cba4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 3 deletions
mmocr/datasets/preparers/dumpers

View File

@ -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.