From 3fbdf930047be9c99c1cc13093f588177a2d0154 Mon Sep 17 00:00:00 2001 From: Martin <maarand@teklia.com> Date: Tue, 1 Feb 2022 11:42:34 +0100 Subject: [PATCH] no more best_classes --- kaldi_data_generator/main.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/kaldi_data_generator/main.py b/kaldi_data_generator/main.py index 657e102..4123088 100644 --- a/kaldi_data_generator/main.py +++ b/kaldi_data_generator/main.py @@ -172,13 +172,13 @@ class HTRDataGenerator: try: accepted_zones = [] for elt in self.api_client.cached_paginate( - "ListElementChildren", id=page_id, with_best_classes=True + "ListElementChildren", id=page_id, with_classes=True ): printed = True - for classification in elt["best_classes"]: + for classification in elt["classes"]: if classification["ml_class"]["name"] == "handwritten": printed = False - for classification in elt["best_classes"]: + for classification in elt["classes"]: if classification["ml_class"]["name"] in self.accepted_classes: if self.should_filter_printed: if not printed: @@ -336,7 +336,7 @@ class HTRDataGenerator: return { elem["id"]: [ best_class["ml_class"]["name"] - for best_class in elem["best_classes"] + for best_class in elem["classes"] if best_class["state"] != "rejected" ] for elem in self.api_client.cached_paginate( @@ -344,7 +344,7 @@ class HTRDataGenerator: id=page_id, recursive=True, type=TEXT_LINE, - with_best_classes=True, + with_classes=True, ) } -- GitLab