Simplify logic

Co-authored-by: Tim Hoffmann <[email protected]>

Author: aitikgupta

lib/matplotlib/cm.py

@@ -367,10 +367,12 @@ def set_array(self, A):
         ----------
         A : array-like or None
         """
-        A = cbook.safe_masked_invalid(A, copy=True) if A is not None else None
-
-        if (A is not None and A.dtype != np.uint8 and
-                not np.can_cast(A.dtype, float, "same_kind")):
+        if A is None:
+            self._A = None
+            return
+            
+        A = cbook.safe_masked_invalid(A, copy=True)
+        if not np.can_cast(A.dtype, float, "same_kind"):
             raise TypeError("Image data of dtype {} cannot be converted to "
                             "float".format(A.dtype))