Don't use shutdown_default_executor for Python 3.8

triton-inference-server · Nov 14, 2024 · 5eabfb0 · 5eabfb0
1 parent f9b2566
commit 5eabfb0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pytriton/proxy/inference.py b/pytriton/proxy/inference.py
@@ -333,7 +333,9 @@ def run(self):
             try:
                 _cancel_all_tasks(self._loop)
                 self._loop.run_until_complete(self._loop.shutdown_asyncgens())
-                self._loop.run_until_complete(self._loop.shutdown_default_executor())
+                # shutdown default executor was introduced in Python 3.9
+                if hasattr(self._loop, "shutdown_default_executor"):
+                    self._loop.run_until_complete(self._loop.shutdown_default_executor())
             finally:
                 asyncio.set_event_loop(None)
                 self._loop.close()