[numpy] Replace uses of np.in1d with np.isin(...).ravel().

np.in1d is removed in NumPy 2.4.

PiperOrigin-RevId: 839920619

Author: hawkinsp

ffn/inference/segmentation.py

@@ -32,7 +32,7 @@ def clear_dust(data: np.ndarray, min_size: int = 10):
   """
   ids, sizes = np.unique(data, return_counts=True)
   small = ids[sizes < min_size]
-  small_mask = np.in1d(data.flat, small).reshape(data.shape)
+  small_mask = np.isin(data.flat, small).ravel().reshape(data.shape)
   data[small_mask] = 0
   return data
 

ffn/inference/storage.py

@@ -391,8 +391,7 @@ def build_mask(masks, corner, subvol_size, mask_volume_map=None,
             src_corner, channel_mask, corner, subvol_size)
 
         if chan_config.values:
-          bool_mask = np.in1d(channel_mask,
-                              chan_config.values).reshape(channel_mask.shape)
+          bool_mask = np.isin(channel_mask, chan_config.values).ravel().reshape(channel_mask.shape)
         else:
           bool_mask = ((channel_mask >= chan_config.min_value) &
                        (channel_mask <= chan_config.max_value))