lint

apache · Oct 8, 2024 · 97faf12 · 97faf12
1 parent fa1fd55
commit 97faf12
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sdks/python/apache_beam/ml/inference/vllm_inference.py b/sdks/python/apache_beam/ml/inference/vllm_inference.py
@@ -36,7 +36,8 @@
 from apache_beam.ml.inference.base import ModelHandler
 from apache_beam.ml.inference.base import PredictionResult
 from apache_beam.utils import subprocess_server
-from openai import AsyncOpenAI, OpenAI
+from openai import AsyncOpenAI
+from openai import OpenAI
 
 try:
   import vllm  # pylint: disable=unused-import