We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7be4aba commit 7600de3Copy full SHA for 7600de3
tests/conftest.py
@@ -1,6 +1,11 @@
1
+import logging
2
+
3
import pytest
4
5
from bioimageio.core import export_resource_package
6
7
+logger = logging.getLogger(__name__)
8
9
# test models for various frameworks
10
torch_models = []
11
torch_models_pre_3_10 = ["unet2d_fixed_shape", "unet2d_multi_tensor", "unet2d_nuclei_broad_model"]
@@ -41,6 +46,7 @@
41
46
import torch
42
47
43
48
torch_version = tuple(map(int, torch.__version__.split(".")[:2]))
49
+ logger.warning(f"detected torch version {torch_version}.x")
44
50
except ImportError:
45
51
torch = None
52
torch_version = None
0 commit comments