sonusai 0.15.4__py3-none-any.whl → 0.15.5__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (5) hide show
  1. sonusai/onnx_predict.py +3 -0
  2. {sonusai-0.15.4.dist-info → sonusai-0.15.5.dist-info}/METADATA +5 -5
  3. {sonusai-0.15.4.dist-info → sonusai-0.15.5.dist-info}/RECORD +5 -5
  4. {sonusai-0.15.4.dist-info → sonusai-0.15.5.dist-info}/WHEEL +0 -0
  5. {sonusai-0.15.4.dist-info → sonusai-0.15.5.dist-info}/entry_points.txt +0 -0
sonusai/onnx_predict.py CHANGED
@@ -223,6 +223,9 @@ def pad_and_predict(feature: Feature,
223
223
  model = rt.InferenceSession(model_name, providers=['CPUExecutionProvider'])
224
224
 
225
225
  predict_arr = np.vstack(predict)
226
+ # Combine [sequences, batch_size, ...] into [frames, ...]
227
+ predict_shape = predict_arr.shape
228
+ predict_arr = np.reshape(predict_arr, [predict_shape[0] * predict_shape[1], *predict_shape[2:]])
226
229
  predict_arr, _ = reshape_outputs(predict=predict_arr, timesteps=timesteps)
227
230
  predict_arr = predict_arr[:frames, :]
228
231
 
{sonusai-0.15.4.dist-info → sonusai-0.15.5.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: sonusai
3
- Version: 0.15.4
3
+ Version: 0.15.5
4
4
  Summary: Framework for building deep neural network models for sound, speech, and voice AI
5
5
  Home-page: https://aaware.com
6
6
  License: GPL-3.0-only
@@ -17,12 +17,12 @@ Classifier: Programming Language :: Python :: 3.11
17
17
  Requires-Dist: PyYAML (>=6.0.1,<7.0.0)
18
18
  Requires-Dist: aixplain (>=0.2.6,<0.3.0)
19
19
  Requires-Dist: dataclasses-json (>=0.6.1,<0.7.0)
20
- Requires-Dist: deepgram-sdk (>=2.11.0,<3.0.0)
20
+ Requires-Dist: deepgram-sdk (>=3.0.0,<4.0.0)
21
21
  Requires-Dist: docopt (>=0.6.2,<0.7.0)
22
22
  Requires-Dist: faster-whisper (>=0.10.0,<0.11.0)
23
23
  Requires-Dist: ffmpeg (>=1.4,<2.0)
24
24
  Requires-Dist: greenlet (>=3.0.1,<4.0.0)
25
- Requires-Dist: grpcio (==1.59.2)
25
+ Requires-Dist: grpcio (==1.60.0)
26
26
  Requires-Dist: jiwer (>=3.0.3,<4.0.0)
27
27
  Requires-Dist: keras-tuner (>=1.4.5,<2.0.0)
28
28
  Requires-Dist: librosa (>=0.10.1,<0.11.0)
@@ -35,7 +35,7 @@ Requires-Dist: pandas (>=2.1.1,<3.0.0)
35
35
  Requires-Dist: pesq (>=0.0.4,<0.0.5)
36
36
  Requires-Dist: pyaaware (>=1.5.3,<2.0.0)
37
37
  Requires-Dist: pydub (>=0.25.1,<0.26.0)
38
- Requires-Dist: pystoi (>=0.3.3,<0.4.0)
38
+ Requires-Dist: pystoi (>=0.4.0,<0.5.0)
39
39
  Requires-Dist: python-magic (>=0.4.27,<0.5.0)
40
40
  Requires-Dist: requests (>=2.31.0,<3.0.0)
41
41
  Requires-Dist: resampy (>=0.4.2,<0.5.0)
@@ -45,7 +45,7 @@ Requires-Dist: sh (>=2.0.6,<3.0.0)
45
45
  Requires-Dist: soundfile (>=0.12.1,<0.13.0)
46
46
  Requires-Dist: sox (>=1.4.1,<2.0.0)
47
47
  Requires-Dist: speechrecognition (>=3.10.0,<4.0.0)
48
- Requires-Dist: tensorflow (>=2.14.0,<3.0.0)
48
+ Requires-Dist: tensorflow (>=2.15.0,<3.0.0)
49
49
  Requires-Dist: tensorflow-addons (>=0.23.0,<0.24.0)
50
50
  Requires-Dist: tf2onnx (>=1.15.1,<2.0.0)
51
51
  Requires-Dist: torch (>=2.1.0,<3.0.0)
{sonusai-0.15.4.dist-info → sonusai-0.15.5.dist-info}/RECORD RENAMED
@@ -69,7 +69,7 @@ sonusai/mixture/truth_functions/sed.py,sha256=8cHjEFjZaH_0hIOHhPmj4AJz2GpEADM6Ys
69
69
  sonusai/mixture/truth_functions/target.py,sha256=V-bXcJZ3P83NHKqYs4RcsxxuZdO8kqzSEN5CNRa5e0k,5394
70
70
  sonusai/mkmanifest.py,sha256=dIPVFKKhnhHdq63OGr6p__pK7fyx3OdKVtbmGUJxsR8,7078
71
71
  sonusai/mkwav.py,sha256=LZNyhq4gJEs_NtGvRsYHA2qfgkkODpt6HoH1b-Tjjuw,5266
72
- sonusai/onnx_predict.py,sha256=FBUXgANCglEwb4Yhg2H7EvhLcWwRFBs5B_rtaBEvIC8,8854
72
+ sonusai/onnx_predict.py,sha256=RhQbbNG3w6rCXuSFUWCaQmUH5JzSP2hmu6TG5_81IVA,9055
73
73
  sonusai/plot.py,sha256=GPrbk7SxwojtjR9KLE9jQ6ywYWXca296W0iaRPdnBHI,16982
74
74
  sonusai/post_spenh_targetf.py,sha256=Y9J5JbFnCFKCednNiYvVXaC5Z3lF5KxifFJ2RCy4jmg,4975
75
75
  sonusai/queries/__init__.py,sha256=oKY5JeqZ4Cz7DwCwPc1_ydB8bUs6KaMcWFp_w02TjOs,255
@@ -122,7 +122,7 @@ sonusai/utils/trim_docstring.py,sha256=dSrtiRsEN4wkkvKBp6WDr13RUypfqZzgH_jOBLs1o
122
122
  sonusai/utils/wave.py,sha256=TKE-CNPGFXNXUW626CBPzCTNgWJut8I0ZEUsgG9q4Po,586
123
123
  sonusai/utils/yes_or_no.py,sha256=eMLXBVH0cEahiXY4W2KNORmwNQ-ba10eRtldh0y4NYg,263
124
124
  sonusai/vars.py,sha256=m2AefF0m5bXWGXpJj8Pi42zWL2ydeEj7bkak3GrtMyM,940
125
- sonusai-0.15.4.dist-info/METADATA,sha256=ozwRzn2oun0k1CULOzzLQo46oGCqqr_4Za8ig5jqw1Y,3097
126
- sonusai-0.15.4.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
127
- sonusai-0.15.4.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
128
- sonusai-0.15.4.dist-info/RECORD,,
125
+ sonusai-0.15.5.dist-info/METADATA,sha256=xNJ4M_KpE7LCYFAPPCJ5Nn6_-vDE_yUY1DsebQT2nXE,3096
126
+ sonusai-0.15.5.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
127
+ sonusai-0.15.5.dist-info/entry_points.txt,sha256=zMNjEphEPO6B3cD1GNpit7z-yA9tUU5-j3W2v-UWstU,92
128
+ sonusai-0.15.5.dist-info/RECORD,,