Changed to cast anything that is not float16, float32, float64

br-Zhang · br-Zhang · commit a453ba4519d6 · 2018-03-13T21:00:08.000-04:00
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -363,9 +363,11 @@ def _make_image(self, A, in_bbox, out_bbox, clip_bbox, magnification=1.0,
                 if inp_dtype.kind == 'f':
                     scaled_dtype = A.dtype
                     # Cast to float64
-                    if A.dtype == np.longdouble:
+                    if not (A.dtype == np.float64 or
+                            A.dtype == np.float32 or
+                            A.dtype == np.float16):
                         warnings.warn(
-                            "Casting input data to float64 for imshow."
+                            'Casting input data to float64 for imshow.'
                         )
                         scaled_dtype = np.float64
                 else:
