[numpy] Fix users of NumPy APIs that are removed in NumPy 2.0.

This change migrates users of APIs removed in NumPy 2.0 to their recommended replacements (https://numpy.org/devdocs/numpy_2_0_migration_guide.html). PiperOrigin-RevId: 656405338
tensorflow · Jul 26, 2024 · 2123db7 · 2123db7
1 parent 2f0e2f3
commit 2123db7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensorflow_datasets/core/features/audio_feature_test.py b/tensorflow_datasets/core/features/audio_feature_test.py
@@ -247,7 +247,7 @@ def _write_audio_file(np_audio, path, file_format):
         f'`{np_audio.dtype}`.'
     )
 
-  data = array.array(np.sctype2char(dtype), np_audio.reshape((-1,)))
+  data = array.array(np.dtype(dtype).char, np_audio.reshape((-1,)))
   sample_width = np.dtype(dtype).alignment
   num_channels = np_audio.shape[1] if len(np_audio.shape) == 2 else 1
 

diff --git a/tensorflow_datasets/image_classification/corruptions.py b/tensorflow_datasets/image_classification/corruptions.py
@@ -132,7 +132,7 @@ def plasma_fractal(mapsize=512, wibbledecay=3):
   """
   if mapsize & (mapsize - 1) != 0:
     raise ValueError('mapsize must be a power of two.')
-  maparray = np.empty((mapsize, mapsize), dtype=np.float_)
+  maparray = np.empty((mapsize, mapsize), dtype=np.float64)
   maparray[0, 0] = 0
   stepsize = mapsize
   wibble = 100