Merge pull request #26807 from oscargus/torchfix

tacaswell · web-flow · commit 3d5585a9558d · 2023-09-18T12:00:57.000-04:00
Catch ValueError to support pytorch (and others) plotting
diff --git a/lib/matplotlib/cbook.py b/lib/matplotlib/cbook.py
@@ -1685,7 +1685,7 @@ def safe_first_element(obj):
 def _safe_first_finite(obj, *, skip_nonfinite=True):
     """
     Return the first finite element in *obj* if one is available and skip_nonfinite is
-    True. Otherwise return the first element.
+    True. Otherwise, return the first element.
 
     This is a method for internal use.
 
@@ -1697,7 +1697,7 @@ def safe_isfinite(val):
             return False
         try:
             return math.isfinite(val)
-        except TypeError:
+        except (TypeError, ValueError):
             pass
         try:
             return np.isfinite(val) if np.isscalar(val) else True
