BUG: Modify match words for catching error message (#37)

xorbitsai · Jul 5, 2023 · de6e1e1 · de6e1e1
1 parent 7d2f12f
commit de6e1e1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/python/xoscar/backends/indigen/pool.py b/python/xoscar/backends/indigen/pool.py
@@ -309,7 +309,7 @@ async def is_sub_pool_alive(self, process: multiprocessing.Process):
  try:
  return await asyncio.to_thread(process.is_alive)
  except RuntimeError as ex: # pragma: no cover
- if "cannot schedule new futures after interpreter shutdown" not in str(ex):
+ if "cannot schedule new futures" not in str(ex):
  # when atexit is triggered, the default pool might be shutdown
  # and to_thread will fail
  raise

diff --git a/python/xoscar/backends/pool.py b/python/xoscar/backends/pool.py
@@ -1408,10 +1408,7 @@ async def monitor_sub_pools(self):
  except asyncio.CancelledError:
  raise
  except RuntimeError as ex: # pragma: no cover
- if (
- "cannot schedule new futures after interpreter shutdown"
- not in str(ex)
- ):
+ if "cannot schedule new futures" not in str(ex):
  # to silence log when process exit, otherwise it
  # will raise "RuntimeError: cannot schedule new futures
  # after interpreter shutdown".