From fd021209f71358e57c6d493a44c71efe67ba9957 Mon Sep 17 00:00:00 2001
From: Martin <maarand@teklia.com>
Date: Wed, 16 Feb 2022 17:12:57 +0100
Subject: [PATCH] fix linting
---
kaldi_data_generator/main.py | 9 ++++++---
1 file changed, 6 insertions(+), 3 deletions(-)
diff --git a/kaldi_data_generator/main.py b/kaldi_data_generator/main.py
index 0b1ff4e..1a6bf9a 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) "
--
GitLab