File tree Expand file tree Collapse file tree 1 file changed +3
-8
lines changed Expand file tree Collapse file tree 1 file changed +3
-8
lines changed Original file line number Diff line number Diff line change 2222
2323import numpy as np
2424import xarray as xr
25+ from bioimageio .spec .model .v0_5 import BATCH_AXIS_ID
2526from loguru import logger
2627from numpy .typing import NDArray
2728from typing_extensions import assert_never
2829
29- from bioimageio .spec .model .v0_5 import BATCH_AXIS_ID
30-
3130from .axis import AxisId , PerAxis
3231from .common import MemberId
3332from .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 )
You can’t perform that action at this time.
0 commit comments