Skip to content

Commit

Permalink
Add Workaround For Windows Users
Browse files Browse the repository at this point in the history
  • Loading branch information
Ma5onic committed Aug 15, 2022
1 parent 18014d7 commit 5f1eb7a
Showing 1 changed file with 12 additions and 8 deletions.
20 changes: 12 additions & 8 deletions evaluator/music_demixing.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
from contextlib import contextmanager
from os import listdir
from os.path import isfile, join
from sys import platform

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

@contextmanager
def time_limit(seconds):
def signal_handler(signum, frame):
raise TimeoutException("Prediction timed out!")

signal.signal(signal.SIGALRM, signal_handler)
signal.alarm(seconds)
try:
if platform != "win32":
def signal_handler(signum, frame):
raise TimeoutException("Prediction timed out!")

signal.signal(signal.SIGALRM, signal_handler)
signal.alarm(seconds)
try:
yield
finally:
signal.alarm(0)
else:
yield
finally:
signal.alarm(0)


class MusicDemixingPredictor:
Expand Down

0 comments on commit 5f1eb7a

Please sign in to comment.