small fix

openvinotoolkit · Oct 16, 2024 · 779cc8f · 779cc8f
1 parent 8104a7c
commit 779cc8f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/otx/core/metrics/average_precision_3d.py b/src/otx/core/metrics/average_precision_3d.py
@@ -45,6 +45,7 @@ def reset(self) -> None:
         super().reset()
         self.preds: list[dict[str, np.array]] = []
         self.targets: list[dict[str, np.array]] = []
+        self.mean_ap.reset()
 
     def update(self, preds: list[dict[str, Tensor]], target: list[dict[str, Tensor]]) -> None:
         """Update total predictions and targets from given batch predicitons and targets."""

diff --git a/tests/integration/cli/test_export_inference.py b/tests/integration/cli/test_export_inference.py
@@ -49,7 +49,7 @@ def fxt_local_seed() -> int:
     "zero_shot_visual_prompting": "test/f1-score",
     "action_classification": "test/accuracy",
     "keypoint_detection": "test/PCK",
-    "object_detection_3d": "test/mAP_bbox_3d",
+    "object_detection_3d": "test/[email protected]",
 }