From 9f19bf970b086e9b79840b08ebeb40681dec2005 Mon Sep 17 00:00:00 2001 From: Can Zhao Date: Thu, 16 Jun 2022 11:41:50 -0400 Subject: [PATCH] corner case in test Signed-off-by: Can Zhao --- tests/test_detection_coco_metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_detection_coco_metrics.py b/tests/test_detection_coco_metrics.py index d5c8445f8e..b139377511 100644 --- 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__":