From e3f1046ebbfdb78ccbaa21cda0912ab56dc754b2 Mon Sep 17 00:00:00 2001 From: dscripka Date: Thu, 18 Jan 2024 20:16:41 -0500 Subject: [PATCH] fix flake8 issue --- openwakeword/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/openwakeword/utils.py b/openwakeword/utils.py index 6572e23..8da8048 100644 --- a/openwakeword/utils.py +++ b/openwakeword/utils.py @@ -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))