Skip to content

Commit 5f1eb7a

Browse files
authored
Add Workaround For Windows Users
1 parent 18014d7 commit 5f1eb7a

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

evaluator/music_demixing.py

+12-8
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from contextlib import contextmanager
1111
from os import listdir
1212
from os.path import isfile, join
13+
from sys import platform
1314

1415
import soundfile as sf
1516
import numpy as np
@@ -21,15 +22,18 @@ class TimeoutException(Exception): pass
2122

2223
@contextmanager
2324
def time_limit(seconds):
24-
def signal_handler(signum, frame):
25-
raise TimeoutException("Prediction timed out!")
26-
27-
signal.signal(signal.SIGALRM, signal_handler)
28-
signal.alarm(seconds)
29-
try:
25+
if platform != "win32":
26+
def signal_handler(signum, frame):
27+
raise TimeoutException("Prediction timed out!")
28+
29+
signal.signal(signal.SIGALRM, signal_handler)
30+
signal.alarm(seconds)
31+
try:
32+
yield
33+
finally:
34+
signal.alarm(0)
35+
else:
3036
yield
31-
finally:
32-
signal.alarm(0)
3337

3438

3539
class MusicDemixingPredictor:

0 commit comments

Comments
 (0)