diff --git a/kaldi_data_generator/main.py b/kaldi_data_generator/main.py index 0b1ff4e3a1fd146c27e173232ca1ef2ea3192f50..1a6bf9afbab81966ff9bca0d16aab84bb7b49764 100644 --- a/kaldi_data_generator/main.py +++ b/kaldi_data_generator/main.py @@ -104,7 +104,9 @@ class HTRDataGenerator: self.extraction_mode = extraction self.accepted_classes = accepted_classes self.ignored_classes = ignored_classes - self.should_filter_by_class = bool(self.accepted_classes) or bool(self.ignored_classes) + self.should_filter_by_class = bool(self.accepted_classes) or bool( + self.ignored_classes + ) self.accepted_worker_version_ids = accepted_worker_version_ids self.should_filter_by_worker = bool(self.accepted_worker_version_ids) self.style = style @@ -852,7 +854,6 @@ def create_parser(): help="List of ignored ml_class names. Filter lines by class", ) - parser.add_argument( "--accepted_classes", nargs="*", @@ -924,7 +925,9 @@ def main(): ) if args.style and (args.accepted_classes or args.ignored_classes): - if set(STYLE_CLASSES) & (set(args.accepted_classes) | set(args.ignored_classes)): + if set(STYLE_CLASSES) & ( + set(args.accepted_classes) | set(args.ignored_classes) + ): parser.error( f"--style class values ({STYLE_CLASSES}) shouldn't be in the accepted_classes list " f"(or ignored_classes list) "