experimenting with error improvements

CHANGELOG.md

@@ -21,6 +21,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Added support for calculating segmentation quality and recognition quality in `PanopticQuality` metric ([#2381](https://github.com/Lightning-AI/torchmetrics/pull/2381))
 
 
+- Added `pretty-errors` for improving error prints ([#2431](https://github.com/Lightning-AI/torchmetrics/pull/2431))
+
+
 ### Changed
 
 - Made `__getattr__` and `__setattr__` of `ClasswiseWrapper` more general ([#2424](https://github.com/Lightning-AI/torchmetrics/pull/2424))

requirements/base.txt

@@ -6,3 +6,4 @@ packaging >17.1
 torch >=1.10.0, <2.3.0
 typing-extensions; python_version < '3.9'
 lightning-utilities >=0.8.0, <0.12.0
+pretty-errors ==1.2.25

src/torchmetrics/__init__.py

@@ -14,6 +14,9 @@
 _PACKAGE_ROOT = os.path.dirname(__file__)
 _PROJECT_ROOT = os.path.dirname(_PACKAGE_ROOT)
 
+if package_available("pretty_errors"):
+    import pretty_errors  # noqa: F401
+
 if package_available("PIL"):
     import PIL