From fa1fd55169bc58830405eaed6c0d3f823fe6ced0 Mon Sep 17 00:00:00 2001 From: Danny McCormick Date: Tue, 8 Oct 2024 09:45:52 -0400 Subject: [PATCH] Remove bad exceptions --- sdks/python/apache_beam/ml/inference/vllm_inference.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/sdks/python/apache_beam/ml/inference/vllm_inference.py b/sdks/python/apache_beam/ml/inference/vllm_inference.py index e44bfb3c604b..88f182b08f10 100644 --- a/sdks/python/apache_beam/ml/inference/vllm_inference.py +++ b/sdks/python/apache_beam/ml/inference/vllm_inference.py @@ -152,7 +152,6 @@ def check_connectivity(self, retries=3): time.sleep(5) if retries == 0: - raise Exception('test') self._server_started = False raise Exception( "Failed to start vLLM server, polling process exited with code " + @@ -161,8 +160,6 @@ def check_connectivity(self, retries=3): else: self.start_server(retries - 1) - raise Exception('test2') - class VLLMCompletionsModelHandler(ModelHandler[str, PredictionResult,