From 39e76d76fd411d64b7481cd21227f32226c124e3 Mon Sep 17 00:00:00 2001 From: NielsRogge <48327001+NielsRogge@users.noreply.github.com> Date: Mon, 1 Aug 2022 15:56:10 +0200 Subject: [PATCH] Fix doc tests (#18397) Co-authored-by: Niels Rogge --- src/transformers/models/detr/modeling_detr.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/detr/modeling_detr.py b/src/transformers/models/detr/modeling_detr.py index c1a164b48..67adaf626 100644 --- a/src/transformers/models/detr/modeling_detr.py +++ b/src/transformers/models/detr/modeling_detr.py @@ -1419,7 +1419,7 @@ class DetrForObjectDetection(DetrPreTrainedModel): ... if score > 0.9: ... print( ... f"Detected {model.config.id2label[label.item()]} with confidence " - ... f"{round(score.item(), 3)} at location {box}." + ... f"{round(score.item(), 3)} at location {box}" ... ) Detected remote with confidence 0.998 at location [40.16, 70.81, 175.55, 117.98] Detected remote with confidence 0.996 at location [333.24, 72.55, 368.33, 187.66]