Skip to content

Commit 792c865

Browse files
claudevdmClaude
andauthored
Re-enable logging after importing vllm. (#35103)
Co-authored-by: Claude <cvandermerwe@google.com>
1 parent c6c815d commit 792c865

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

sdks/python/apache_beam/ml/inference/vllm_inference.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@
4343
os.environ["VLLM_CONFIGURE_LOGGING"] = "0"
4444
import vllm # pylint: disable=unused-import
4545
logging.info('vllm module successfully imported.')
46+
os.environ["VLLM_CONFIGURE_LOGGING"] = "1"
4647
except ModuleNotFoundError:
4748
msg = 'vllm module was not found. This is ok as long as the specified ' \
4849
'runner has vllm dependencies installed.'

0 commit comments

Comments
 (0)