yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-10-30 17:29:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix resource limit issue (#589)

Browse Source
This commit is contained in:
Chi Wang 2022-06-15 13:46:52 -07:00 committed by GitHub
parent 79a24d06a9
commit cf1dfd3966
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
flaml/model.py
View File

@ -56,7 +56,11 @@ def limit_resource(memory_limit, time_limit):
if memory_limit > 0: if memory_limit > 0:
soft, hard = resource.getrlimit(resource.RLIMIT_AS) soft, hard = resource.getrlimit(resource.RLIMIT_AS)
if soft < 0 and (hard < 0 or memory_limit <= hard) or memory_limit < soft: if soft < 0 and (hard < 0 or memory_limit <= hard) or memory_limit < soft:
resource.setrlimit(resource.RLIMIT_AS, (memory_limit, hard)) try:
resource.setrlimit(resource.RLIMIT_AS, (int(memory_limit), hard))
except ValueError:
# According to https://bugs.python.org/issue40518, it's a mac-specific error.
pass
main_thread = False main_thread = False
if time_limit is not None: if time_limit is not None:
try: try: