Skip to content

Commit 60edb28

Browse files
sdks/python: fix linting issues
1 parent 973a633 commit 60edb28

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

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

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,6 @@
1818

1919
from typing import Optional
2020

21-
import openai
22-
2321
import apache_beam as beam
2422
from apache_beam.ml.inference.base import RunInference
2523
from apache_beam.ml.rag.embeddings.base import create_rag_adapter

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def request(
103103
) -> Iterable:
104104
"""Makes a request to OpenAI embedding API and returns embeddings."""
105105
# Prepare arguments for the API call
106-
kwargs = {
106+
kwargs: dict[str, Any] = {
107107
"model": self.model_name,
108108
"input": batch,
109109
}

0 commit comments

Comments
 (0)