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