sdks/python: fix linting issues

Author: moawnallah

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

@@ -31,12 +31,13 @@
 from typing import Any
 from typing import Optional
 
+from openai import AsyncOpenAI
+from openai import OpenAI
+
 from apache_beam.io.filesystems import FileSystems
 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
-from openai import OpenAI
 
 try:
   # VLLM logging config breaks beam logging.

sdks/python/apache_beam/ml/rag/embeddings/huggingface_test.py

@@ -24,14 +24,14 @@
 
 import apache_beam as beam
 from apache_beam.ml.rag.embeddings.huggingface import HuggingfaceTextEmbeddings
+from apache_beam.ml.rag.embeddings.test_utils import chunk_approximately_equals
 from apache_beam.ml.rag.types import Chunk
 from apache_beam.ml.rag.types import Content
 from apache_beam.ml.rag.types import Embedding
 from apache_beam.ml.transforms.base import MLTransform
 from apache_beam.testing.test_pipeline import TestPipeline
 from apache_beam.testing.util import assert_that
 from apache_beam.testing.util import equal_to
-from apache_beam.ml.rag.embeddings.test_utils import chunk_approximately_equals
 
 # pylint: disable=unused-import
 try:

sdks/python/apache_beam/ml/rag/embeddings/open_ai.py

@@ -24,8 +24,7 @@
 from apache_beam.ml.rag.types import Chunk
 from apache_beam.ml.transforms.base import EmbeddingsManager
 from apache_beam.ml.transforms.base import _TextEmbeddingHandler
-from apache_beam.ml.transforms.embeddings.open_ai import (
-    _OpenAITextEmbeddingHandler, )
+from apache_beam.ml.transforms.embeddings.open_ai import _OpenAITextEmbeddingHandler
 
 __all__ = ['OpenAITextEmbeddings']
 

sdks/python/apache_beam/ml/rag/embeddings/open_ai_test.py

@@ -19,15 +19,15 @@
 import unittest
 
 import apache_beam as beam
+from apache_beam.ml.rag.embeddings.open_ai import OpenAITextEmbeddings
+from apache_beam.ml.rag.embeddings.test_utils import chunk_approximately_equals
 from apache_beam.ml.rag.types import Chunk
 from apache_beam.ml.rag.types import Content
 from apache_beam.ml.rag.types import Embedding
 from apache_beam.ml.transforms.base import MLTransform
 from apache_beam.testing.test_pipeline import TestPipeline
 from apache_beam.testing.util import assert_that
 from apache_beam.testing.util import equal_to
-from apache_beam.ml.rag.embeddings.open_ai import OpenAITextEmbeddings
-from apache_beam.ml.rag.embeddings.test_utils import chunk_approximately_equals
 
 
 class OpenAITextEmbeddingsTest(unittest.TestCase):

sdks/python/apache_beam/ml/rag/embeddings/vertex_ai_test.py

@@ -21,14 +21,14 @@
 import unittest
 
 import apache_beam as beam
+from apache_beam.ml.rag.embeddings.test_utils import chunk_approximately_equals
 from apache_beam.ml.rag.types import Chunk
 from apache_beam.ml.rag.types import Content
 from apache_beam.ml.rag.types import Embedding
 from apache_beam.ml.transforms.base import MLTransform
 from apache_beam.testing.test_pipeline import TestPipeline
 from apache_beam.testing.util import assert_that
 from apache_beam.testing.util import equal_to
-from apache_beam.ml.rag.embeddings.test_utils import chunk_approximately_equals
 
 # pylint: disable=ungrouped-imports
 try:

sdks/python/apache_beam/ml/transforms/embeddings/open_ai.py

@@ -21,16 +21,17 @@
 from typing import TypeVar
 from typing import Union
 
-import apache_beam as beam
 import openai
+from openai import APIError
+from openai import RateLimitError
+
+import apache_beam as beam
 from apache_beam.ml.inference.base import RemoteModelHandler
 from apache_beam.ml.inference.base import RunInference
 from apache_beam.ml.transforms.base import EmbeddingsManager
 from apache_beam.ml.transforms.base import _TextEmbeddingHandler
 from apache_beam.pvalue import PCollection
 from apache_beam.pvalue import Row
-from openai import APIError
-from openai import RateLimitError
 
 __all__ = ["OpenAITextEmbeddings"]