Skip to content

Commit fb40f40

Browse files
committed
update typing
1 parent 75d55db commit fb40f40

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

src/bioimageio/core/stat_calculators.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,11 @@
2222

2323
import numpy as np
2424
import xarray as xr
25+
from bioimageio.spec.model.v0_5 import BATCH_AXIS_ID
2526
from loguru import logger
2627
from numpy.typing import NDArray
2728
from typing_extensions import assert_never
2829

29-
from bioimageio.spec.model.v0_5 import BATCH_AXIS_ID
30-
3130
from .axis import AxisId, PerAxis
3231
from .common import MemberId
3332
from .sample import Sample
@@ -140,13 +139,9 @@ def compute(
140139
n = int(np.prod([tensor.sizes[d] for d in self._axes]))
141140

142141
if xr.__version__.startswith("2023"):
143-
var = ( # pyright: ignore[reportUnknownVariableType]
144-
xr.dot(c, c, dims=self._axes) / n
145-
)
142+
var = xr.dot(c, c, dims=self._axes) / n
146143
else:
147-
var = ( # pyright: ignore[reportUnknownVariableType]
148-
xr.dot(c, c, dim=self._axes) / n
149-
)
144+
var = xr.dot(c, c, dim=self._axes) / n
150145

151146
assert isinstance(var, xr.DataArray)
152147
std = np.sqrt(var)

0 commit comments

Comments
 (0)