akadmin/openwakeword
1
0
Fork 0
mirror of https://github.com/dscripka/openWakeWord.git synced 2026-08-27 18:17:20 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix flake8 issue

Browse source
This commit is contained in:
dscripka 2024-01-18 20:16:41 -05:00 • committed by GitHub
parent db418ab9f3
commit e3f1046ebb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
openwakeword/utils.py
View file

@ -269,7 +269,7 @@ class AudioFeatures():
result = self._get_melspectrogram(batch)
elif pool:
chunksize = batch.shape[0]//ncpu if batch.shape[0]>=ncpu else 1
chunksize = batch.shape[0]//ncpu if batch.shape[0] >= ncpu else 1
result = np.array(pool.map(self._get_melspectrogram,
batch, chunksize=chunksize))
@ -331,7 +331,7 @@ class AudioFeatures():
result = self.embedding_model_predict(batch)
elif pool:
chunksize = batch.shape[0]//ncpu if batch.shape[0]>=ncpu else 1
chunksize = batch.shape[0]//ncpu if batch.shape[0] >= ncpu else 1
result = np.array(pool.map(self._get_embeddings_from_melspec,
batch, chunksize=chunksize))