From 28978c83fb05ba457d3a94f59023a712debada97 Mon Sep 17 00:00:00 2001 From: Jirka Date: Tue, 5 Mar 2024 21:37:49 +0100 Subject: [PATCH 1/2] experimenting with error improvements --- requirements/base.txt | 1 + src/torchmetrics/__init__.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/requirements/base.txt b/requirements/base.txt index d0286421e42..97c0fdb85a0 100644 --- a/requirements/base.txt +++ b/requirements/base.txt @@ -6,3 +6,4 @@ packaging >17.1 torch >=1.10.0, <=2.2.1 typing-extensions; python_version < '3.9' lightning-utilities >=0.8.0, <0.11.0 +pretty-errors ==1.2.25 diff --git a/src/torchmetrics/__init__.py b/src/torchmetrics/__init__.py index b1549dfaf8b..c1bff7ca459 100644 --- a/src/torchmetrics/__init__.py +++ b/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 From 791cf0df2db0882ab38779e0869edc438d1f240d Mon Sep 17 00:00:00 2001 From: Jirka Date: Thu, 28 Mar 2024 22:26:34 +0100 Subject: [PATCH 2/2] chlog --- CHANGELOG.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/CHANGELOG.md b/CHANGELOG.md index 537d5cf61f7..d4e12300c75 100644 --- a/CHANGELOG.md +++ b/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))