do not overwrite input, use specialized function

damask-multiphysics · Nov 20, 2023 · c0a97a9 · c0a97a9
1 parent 39dacbc
commit c0a97a9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/damask/_colormap.py b/python/damask/_colormap.py
@@ -305,11 +305,11 @@ def shade(self,
         if abs(delta := r-l) * 1e8 <= (avg := 0.5*abs(r+l)):                                        # delta is similar to numerical noise
             l,r = (l-0.5*avg*np.sign(delta),r+0.5*avg*np.sign(delta))                               # extend range to have actual data centered within
 
-        field[np.isnan(field)] = (l+r)/2
+        field_ = np.nan_to_num(field, nan=(l+r)/2, posinf=r, neginf=l)
 
         return Image.fromarray(
             (np.dstack((
-                        self.colors[np.round(np.clip((field-l)/(r-l),0.0,1.0)*(self.N-1)).astype(np.uint16),:3],
+                        self.colors[np.round(np.clip((field_-l)/(r-l),0.0,1.0)*(self.N-1)).astype(np.uint16),:3],
                         mask.astype(float)
                        )
                       )*255