Hide GeneratorExit in _iterate_inference_steps()

bigscience-workshop · Jul 22, 2023 · f150932 · f150932
1 parent eb0664b
commit f150932
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/petals/server/handler.py b/src/petals/server/handler.py
@@ -347,7 +347,7 @@ async def _iterate_inference_steps(
                         anext_task.cancel()
                         get_push_task.cancel()
                         return
-        except:
+        except Exception:
             logger.warning("rpc_inference._iterate_inference_steps() exception:", exc_info=True)
             raise