diff --git a/requirements/fabric/test.txt b/requirements/fabric/test.txt index 2da6ae8854d64..4ce2ff40f5bce 100644 --- a/requirements/fabric/test.txt +++ b/requirements/fabric/test.txt @@ -7,4 +7,3 @@ pytest-rerunfailures ==12.0 pytest-random-order ==1.1.0 click ==8.1.7 tensorboardX >=2.2, <2.7.0 # min version is set by torch.onnx missing attribute -torchmetrics >=0.7.0, <1.5.0 # needed for using fixed compare_version diff --git a/requirements/pytorch/base.txt b/requirements/pytorch/base.txt index cdf3cc03e2985..7a87aa86278c4 100644 --- a/requirements/pytorch/base.txt +++ b/requirements/pytorch/base.txt @@ -5,7 +5,7 @@ torch >=2.1.0, <2.6.0 tqdm >=4.57.0, <4.67.0 PyYAML >=5.4, <6.1.0 fsspec[http] >=2022.5.0, <2024.4.0 -torchmetrics >=0.7.0, <1.5.0 # needed for using fixed compare_version +torchmetrics >=0.7.0, <1.5.0 packaging >=20.0, <=23.1 typing-extensions >=4.4.0, <4.11.0 lightning-utilities >=0.10.0, <0.12.0 diff --git a/src/lightning/fabric/utilities/spike.py b/src/lightning/fabric/utilities/spike.py index 04c554461c58c..e96eccd75b1a2 100644 --- a/src/lightning/fabric/utilities/spike.py +++ b/src/lightning/fabric/utilities/spike.py @@ -52,7 +52,7 @@ def __init__( from torchmetrics.aggregation import MeanMetric from torchmetrics.wrappers import Running else: - raise RuntimeError("SpikeDetection requires torchmetrics>=1.0.0! Please upgrade your version!") + raise RuntimeError("SpikeDetection requires `torchmetrics>=1.0.0` Please upgrade your version.") super().__init__() self.last_val: Union[torch.Tensor, float] = 0.0