Skip to content

Commit 5f2d837

Browse files
committed
fix model inference tolerance reporting
1 parent f1e0782 commit 5f2d837

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

src/bioimageio/core/_resource_tests.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,6 @@
2424
)
2525

2626
import numpy as np
27-
from loguru import logger
28-
from numpy.typing import NDArray
29-
from typing_extensions import NotRequired, TypedDict, Unpack, assert_never, get_args
30-
31-
from bioimageio.core import __version__
3227
from bioimageio.spec import (
3328
AnyDatasetDescr,
3429
AnyModelDescr,
@@ -66,6 +61,11 @@
6661
ValidationSummary,
6762
WarningEntry,
6863
)
64+
from loguru import logger
65+
from numpy.typing import NDArray
66+
from typing_extensions import NotRequired, TypedDict, Unpack, assert_never, get_args
67+
68+
from bioimageio.core import __version__
6969

7070
from ._prediction_pipeline import create_prediction_pipeline
7171
from .axis import AxisId, BatchSize
@@ -816,9 +816,7 @@ def add_warning_entry(msg: str):
816816
r_expected = expected_np[r_max_idx].item()
817817

818818
# Calculate the max absolute difference with the relative tolerance subtracted
819-
abs_diff_wo_rtol: NDArray[np.float32] = (abs_diff - rtol_value).max(
820-
initial=0
821-
)
819+
abs_diff_wo_rtol: NDArray[np.float32] = abs_diff - rtol_value
822820
a_max_idx = np.unravel_index(
823821
abs_diff_wo_rtol.argmax(), abs_diff_wo_rtol.shape
824822
)

0 commit comments

Comments
 (0)