diff --git a/sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py b/sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py
index 0b356e039930..f37f297530ce 100644
--- a/sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py
+++ b/sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py
@@ -380,7 +380,7 @@ def pre_milvus_enrichment() -> VectorDBContainerInfo:
           uri=db.uri,
           user=db.user,
           password=db.password,
-          db_id=db.id,
+          db_name=db.id,
           token=db.token)
       collection_name = MilvusTestHelpers.initialize_db_with_data(
           connection_params)
diff --git a/sdks/python/apache_beam/ml/inference/base.py b/sdks/python/apache_beam/ml/inference/base.py
index 62c4047a961f..84d68ef6c061 100644
--- a/sdks/python/apache_beam/ml/inference/base.py
+++ b/sdks/python/apache_beam/ml/inference/base.py
@@ -573,7 +573,7 @@ def __init__(self, mh_map: dict[str, ModelHandler]):
     # Map key for a model to a unique tag that will persist for the life of
     # that model in memory. A new tag will be generated if a model is swapped
     # out of memory and reloaded.
-    self._tag_map: dict[str, str] = OrderedDict()
+    self._tag_map: OrderedDict[str, str] = OrderedDict()
     # Map a tag to a multiprocessshared model object for that tag. Each entry
     # of this map should last as long as the corresponding entry in _tag_map.
     self._proxy_map: dict[str, multi_process_shared.MultiProcessShared] = {}
diff --git a/sdks/python/pyproject.toml b/sdks/python/pyproject.toml
index d8f09d7474b4..a0c316f1b146 100644
--- a/sdks/python/pyproject.toml
+++ b/sdks/python/pyproject.toml
@@ -212,5 +212,4 @@ not-callable = "ignore"
 invalid-argument = "ignore"
 invalid-inheritance = "ignore"
 not-iterable = "ignore"
-unexpected-keyword = "ignore"
 bad-class-definition = "ignore"