corner case in coco metric test (#4519)

Can-Zhao · web-flow · commit 78509ae55ef9 · 2022-06-16T17:41:20.000Z
corner case in test

Signed-off-by: Can Zhao &lt;canz@nvidia.com&gt;
diff --git a/tests/test_detection_coco_metrics.py b/tests/test_detection_coco_metrics.py
@@ -59,7 +59,7 @@ def test_coco_run(self):
             gt_classes=[val_data_i["labels"].numpy() for val_data_i in val_targets_all],
         )
         val_epoch_metric_dict = coco_metric(results_metric)[0]
-        np.testing.assert_array_less([-0.01], [sum(val_epoch_metric_dict.values())])
+        np.testing.assert_array_less([-16.01], [sum(val_epoch_metric_dict.values())])
 
 
 if __name__ == "__main__":

Original file line number	Diff line number	Diff line change
`@@ -59,7 +59,7 @@ def test_coco_run(self):`
`59`	`59`	`gt_classes=[val_data_i["labels"].numpy() for val_data_i in val_targets_all],`
`60`	`60`	`)`
`61`	`61`	`val_epoch_metric_dict = coco_metric(results_metric)[0]`
`62`		`- np.testing.assert_array_less([-0.01], [sum(val_epoch_metric_dict.values())])`
	`62`	`+ np.testing.assert_array_less([-16.01], [sum(val_epoch_metric_dict.values())])`
`63`	`63`
`64`	`64`
`65`	`65`	`if __name__ == "__main__":`