From cce8636fdf4575148b79da447cb8dc0d4a0092ac Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?F=C3=A9lix=20Voituret?= <fvoituret@deezer.com>
Date: Wed, 6 Nov 2019 18:42:52 +0100
Subject: [PATCH] fix: update writing process

---
 spleeter/commands/separate.py  | 12 +++++-------
 spleeter/utils/audio/ffmpeg.py |  2 +-
 2 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/spleeter/commands/separate.py b/spleeter/commands/separate.py
index 0098351..25b9226 100644
--- a/spleeter/commands/separate.py
+++ b/spleeter/commands/separate.py
@@ -129,7 +129,6 @@ def process_audio(
         yield_single_examples=False)
     # initialize pool for audio export
     pool = Pool(16)
-    tasks = []
     for sample in prediction:
         sample_filename = sample.pop('audio_id', 'unknown_filename').decode()
         input_directory, input_filename = split(sample_filename)
@@ -144,13 +143,12 @@ def process_audio(
                 output_path,
                 output_dirname,
                 f'{instrument}.{codec}')
-            tasks.append(
-                pool.apply_async(
-                    audio_adapter.save,
-                    (filename, waveform, sample_rate, codec)))
+            pool.apply_async(
+                audio_adapter.save,
+                (filename, waveform, sample_rate, codec))
     # Wait for everything to be written
-    for task in tasks:
-        task.wait(timeout=20)
+    pool.join()
+    pool.close()
 
 
 def entrypoint(arguments, params):
diff --git a/spleeter/utils/audio/ffmpeg.py b/spleeter/utils/audio/ffmpeg.py
index 9126eaa..feda039 100644
--- a/spleeter/utils/audio/ffmpeg.py
+++ b/spleeter/utils/audio/ffmpeg.py
@@ -113,7 +113,7 @@ class FFMPEGProcessAudioAdapter(AudioAdapter):
             .output(path, format='f32le', **output_kwargs)
             .run_async(pipe_stdin=True, quiet=True))
         try:
-            process.stdin.write(data.astype('<f4').tostring())
+            process.stdin.write(data.astype('<f4').tobytes())
             process.stdin.close()
             process.wait()
         except IOError: