Skip to content

Commit 7600de3

Browse files
committed
log detected torch_version
1 parent 7be4aba commit 7600de3

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/conftest.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,11 @@
1+
import logging
2+
13
import pytest
4+
25
from bioimageio.core import export_resource_package
36

7+
logger = logging.getLogger(__name__)
8+
49
# test models for various frameworks
510
torch_models = []
611
torch_models_pre_3_10 = ["unet2d_fixed_shape", "unet2d_multi_tensor", "unet2d_nuclei_broad_model"]
@@ -41,6 +46,7 @@
4146
import torch
4247

4348
torch_version = tuple(map(int, torch.__version__.split(".")[:2]))
49+
logger.warning(f"detected torch version {torch_version}.x")
4450
except ImportError:
4551
torch = None
4652
torch_version = None

0 commit comments

Comments
 (0)