From e041a64104db8d2ddf3e33a55a87039a1519acfc Mon Sep 17 00:00:00 2001 From: Sound Separation Team Date: Tue, 30 Jul 2024 16:08:26 +0000 Subject: [PATCH] Fix numpy usages removed in numpy 2.0. PiperOrigin-RevId: 657614534 --- models/train/data_meeting_io.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/train/data_meeting_io.py b/models/train/data_meeting_io.py index 8bda1bb..060cfab 100644 --- a/models/train/data_meeting_io.py +++ b/models/train/data_meeting_io.py @@ -328,7 +328,7 @@ def utterance_info_generator(): (num_utterances, 1), (num_utterances, 1), (num_utterances, 1)] - tensor_type = [np.string_, np.int32, np.int32, np.float32, + tensor_type = [np.bytes_, np.int32, np.int32, np.float32, np.int32, np.float32] (wav_paths, diarizations, spkindices, meeting_scales, start_samples,