experimenting with error improvements

Lightning-AI · Mar 5, 2024 · e97f1c6 · e97f1c6
1 parent 9d76f3f
commit e97f1c6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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