diff options
Diffstat (limited to 'src/text_recognizer/datasets/util.py')
| -rw-r--r-- | src/text_recognizer/datasets/util.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/text_recognizer/datasets/util.py b/src/text_recognizer/datasets/util.py index 321bc67..76bd85f 100644 --- a/src/text_recognizer/datasets/util.py +++ b/src/text_recognizer/datasets/util.py @@ -46,6 +46,7 @@ def fetch_data_loaders( """ def check_dataset_args(args: Dict, split: str) -> Dict: + """Adds train flag to the dataset args.""" args["train"] = True if split == "train" else False return args |