diff options
Diffstat (limited to 'tests/test_onnx_only_infer.py')
| -rw-r--r-- | tests/test_onnx_only_infer.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tests/test_onnx_only_infer.py b/tests/test_onnx_only_infer.py index 69c72be..d41e7fd 100644 --- a/tests/test_onnx_only_infer.py +++ b/tests/test_onnx_only_infer.py @@ -187,9 +187,7 @@ for e in entities: conf = round(float(np.mean(e["probs"])), 3) typ = e["type"] if typ in ("timestamp", "speaker"): - ent_text = ( - fix_timestamp(ent_text) if typ == "timestamp" else fix_speaker(ent_text) - ) + ent_text = fix_timestamp(ent_text) if typ == "timestamp" else fix_speaker(ent_text) out["metadata"][typ] = ent_text else: out["content"].append({"type": typ, "content": ent_text, "confidence": conf}) |