Skip to content

Commit eacbf0a

Browse files
committed
bypass for RUSAGE so it never kills the process
1 parent a82e7e9 commit eacbf0a

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

src/utils/middleware.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,10 @@ def _diff_usages(cls, start, end=None):
7676
@staticmethod
7777
def _get_usage():
7878
try:
79-
utime, stime, *_ = resource.getrusage(resource.RUSAGE_THREAD)
80-
except AttributeError:
81-
utime, stime, *_ = resource.getrusage(resource.RUSAGE_SELF)
82-
return (time.time(), utime, stime)
79+
try:
80+
utime, stime, *_ = resource.getrusage(resource.RUSAGE_THREAD)
81+
except AttributeError:
82+
utime, stime, *_ = resource.getrusage(resource.RUSAGE_SELF)
83+
return (time.time(), utime, stime)
84+
except Exception:
85+
return (time.time(), 0, 0)

0 commit comments

Comments
 (0)