From 28e369bd13605702c099406d096a6bf040398f10 Mon Sep 17 00:00:00 2001 From: Claude Date: Fri, 30 May 2025 19:59:50 +0000 Subject: [PATCH] Re-enable logging after importing vllm. --- sdks/python/apache_beam/ml/inference/vllm_inference.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sdks/python/apache_beam/ml/inference/vllm_inference.py b/sdks/python/apache_beam/ml/inference/vllm_inference.py index c1f94259aca6..c4d5aa2dcd37 100644 --- a/sdks/python/apache_beam/ml/inference/vllm_inference.py +++ b/sdks/python/apache_beam/ml/inference/vllm_inference.py @@ -43,6 +43,7 @@ os.environ["VLLM_CONFIGURE_LOGGING"] = "0" import vllm # pylint: disable=unused-import logging.info('vllm module successfully imported.') + os.environ["VLLM_CONFIGURE_LOGGING"] = "1" except ModuleNotFoundError: msg = 'vllm module was not found. This is ok as long as the specified ' \ 'runner has vllm dependencies installed.'